fixed some tests
minor fix