Fixed silly broken unit test