fixed bug in test