Fix a test assuming CPU mode without explicitly setting it.