Update test to reflect new controller output