Added another test and fixed a bug