Fix tests failing to build.