0
@@ -80,8 +80,7 @@
0
         raise ArgumentError, "Only one of --revision or --commit can be given.  Received both."
0
       end
0
 
0
-      cmd = Piston::Commands::Import.new(:lock => params["lock"].value,
0
-                                         :verbose => params["verbose"].value,
0
+      cmd = Piston::Commands::Import.new(:verbose => params["verbose"].value,
0
                                          :quiet => params["quiet"].value,
0
                                          :force => params["force"].value,
0
                                          :dry_run => params["dry-run"].value,
0
@@ -101,6 +100,13 @@
0
         end
0
         exit_failure!
0
       end
0
+
0
+      # Lock the working copy, if the user asked for it
0
+      cmd = Piston::Commands::LockUnlock.new(:verbose => params["verbose"].value,
0
+                                             :quiet => params["quiet"].value,
0
+                                             :force => params["force"].value,
0
+                                             :dry_run => params["dry-run"].value)
0
+      cmd.run(params["directory"].value, true) if params["lock"].value
0
     end
0
   end
0