Whoops, forgot a class