Fixed wrong inputs in the tests