Fix version flag test