Unable to Locate element using headless HtmlUnitDriver

Tried running my tests using HtmlUnitDriver today but all of them failed, since it was unable to locate elements.

Tried researching and found out many people have faced such issues but were unable to give any specific answer to the query.

When I use FirefoxDriver(), test run and PASS properly.

Any insight as to what could be the issue.

here is the code that I am using:

WebDriver driver = new HtmlUnitDriver(true);

So its JS enabled also.

I added the waits also, but still it does not seem to work