update tests to match #10