Fix build breakage after controller API change