Pressing widget button with mouse click lose keyboard Enhanced input.

Unchecking “IsFocusable” on the widget button worked for me.

Found here: InputAxis stops working after clicking Widget button