fix #2, moar tests