small controller change