Fixed issue with key presses not working