fixed broken unit tests