Fix tests failing on jenkins