Fix for broken test after #65

This fixes the test that PR #65 broke.