Update controller for tests