small controller fix