Fixing issue with taking screenshot of failing test