Fix screenshot code handling