Makes the screenshot work after stopping