fix tests
add test.
heh