Forgot to add the new base class