More unit tests

fixed broken unit test