Fix screenshot not showing up...

Fix for screenshot issue...