Corrected tests to match expectations.

References #1