Fix mouse following functionality