fixed tests
fixed tests