fix tests failing issue