Fixed the controller issue.