corrected broken unit test