Fix warning on controller