Fix broken unit test.
Merge pull request #270 from esheffield/fix_unit_test

Fix broken unit test.