Fixed broken test.  Closes #159