tests are failing
now it works :D