#44 Fix tests, doh!