#2 fixed bug in test classes