and another fix. I don't know unit testing :)