fix some screenshot bugs