Update mock controller helper