fixed tests
compile fix