diff --git a/src/main.cpp b/src/main.cpp
index c53b33a..01edab6 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -2,12 +2,12 @@
 
 int main(int argc, const char** argv)
 {
-  if (argc != 2)
+  if (argc != 3)
     throw std::runtime_error(fmt::format("Usage: {} task config.yaml", argv[0]));
   if (argv[1] == std::string("unfold"))
-    ufo::UnfoldSolver{argv[1]}();
+    ufo::UnfoldSolver{argv[2]}();
   else if (argv[1] == std::string("plot"))
-    ufo::PlotSolver{argv[1]}();
+    ufo::PlotSolver{argv[2]}();
   else
     throw std::runtime_error(fmt::format("Unknown task: {}", argv[1]));
 }