fix tests, did you forget to run them @maethu?