add test fixes