refactor controllers
Small fix