Fix a function, add more tests