Fix unit test
Merge pull request #157 from elandau/master

Fix unit test