Fixed test from prev commit. Fix for issue #150