Set the controller properly