Fixing controller error.: