diff --git a/Modules/Filtering/ImageCompare/test/itkSTAPLEImageFilterTest.cxx b/Modules/Filtering/ImageCompare/test/itkSTAPLEImageFilterTest.cxx index 8652a9061bc..04b944c385f 100644 --- a/Modules/Filtering/ImageCompare/test/itkSTAPLEImageFilterTest.cxx +++ b/Modules/Filtering/ImageCompare/test/itkSTAPLEImageFilterTest.cxx @@ -183,8 +183,6 @@ Stapler::Execute() int itkSTAPLEImageFilterTest(int argc, char * argv[]) { - StaplerBase * stapler; - if (argc < 5) { std::cerr << "Use: " << itkNameOfTestExecutableMacro(argv) @@ -194,6 +192,8 @@ itkSTAPLEImageFilterTest(int argc, char * argv[]) return EXIT_FAILURE; } + StaplerBase * stapler; + if (std::stoi(argv[1]) == 2) { stapler = new Stapler<2>;