Add test for return input when not in special case