[ADD] test fixes