Add more condition to disable the input