fixing controller reference