@@ -253,13 +253,18 @@ process fn (RmProof n')
253253 let ms = idris_metavars i
254254 put $ i { idris_metavars = n : ms }
255255
256−process fn (AddProof n')
256+process fn (AddProof prf)
257257 = do let fb = fn ++ "~"
258258 liftIO $ copyFile fn fb -- make a backup in case something goes wrong!
259259 prog <- liftIO $ readFile fb
260260 i <- get
261− n <- resolveProof n'
262261 let proofs = proof_list i
262+ n' <- case prf of
263+ Nothing -> case proofs of
264+ [] -> fail "No proof to add"
265+ ((x, p) : _) -> return x
266+ Just nm -> return nm
267+ n <- resolveProof n'
263268 case lookup n proofs of
264269 Nothing -> iputStrLn "No proof to add"
265270 Just p -> do let prog' = insertScript (showProof (lit fn) n p) ls