Fix missing import and bad query for screenshots