Fix screenshot for firefox. Closes #311