minor fix with input