Fix unit tests when running on ARM64