Fixed up a whoops