Forgot to add the new class... shhh