Fix grid screenshot