fix the if statement for screenshot to reflect the correct path