Fix screenshot method