minor fix, verified that updated tests all pass.