Respect the new flag in tests. Fixes #12