bug/test fixes