Fix a unit test because of #141

Fixed unit test

closes #148