update tests, now it works