Fixed mocked driver screenshot taking