Forget to reference models on main class