add minor input error check