Fixed the input tests