Fix escape key not working