fix tests failing