Fix keyword input bug