Fix tests for #10