Fixed minor controller issue