Stupid, forgot to add a whole class