updated tests to account for slight api changes