Update tests mock controller helper