#11 fixed bug in test