Fixed the unit test to do what it claims