firefox-headless contol box is missing from screenshot - selenium - linux

Question

I am facing exactly same problem https://github.com/mozilla/geckodriver/issues/1358.

Firefox-headless mode screenshots taken from linux are missing input box