fix #11

we have tests yay!