Fixed an issue with input types