Simplified cod in the unit test.

References #693.