Update tests to work with previous commit. #29