just small controller fix ;)