Fixing a test that works locally but not on the build server