Fixing controller to make PR pass