fixed broken unit test