fixing mistake, untested part wasnt working... next time I should use tdd