fixed tests

okay, I'm agree