Ref #106 - Fixing test failure