Fixing controller tests.