Fix controller tests.