fixes #2

only the tests had to be rewritten

the code was allready working so it supports both ways