Forgot to add the entity class