Improve tests due to #20, but all tests pass, it's not a bug