#76 fixed broken unit test