#157 test fixes 2