Fixed screenshot problem

Was due to a dumb function (didn't use the argument)