updated tests (fixed bug)