Fix for direction input