Fix tests mock controller helper