Try adding a controller class if needed.