fixed tests (missed in previous commit)