Woops, forgot a class