Fix screenshot to show error message