test fixes

test fixes