Fixing Controller Reference