Little controller fix