fixed tests
small fixes