Fixes broken controller tests.