Fixed unit test failure

Failure is caused by commit 2913065 [GH-146]