Remove references to controller which broke the build