Improved unit test on input action