Fix the build by wrapping the test in run