Fix: bug with screenshots