1 files changed, 7 insertions, 2 deletions

diff --git a/src/luarocks/install.lua b/src/luarocks/install.lua
index 3b399ebc..8e8b5f6b 100644
--- a/src/luarocks/install.lua
+++ b/src/luarocks/install.lua
@@ -64,10 +64,15 @@ function install_binary_rock(rock_file)
       if err then return nil, err end
    end
 
-   ok, err = rep.run_hook(rockspec, "post_install")
+   ok, err = rep.deploy_files(name, version)
    if err then return nil, err end
 
-   ok, err = rep.deploy_files(name, version)
+   util.remove_scheduled_function(rollback)
+   rollback = util.schedule_function(function()
+      rep.delete_version(name, version)
+   end)
+
+   ok, err = rep.run_hook(rockspec, "post_install")
    if err then return nil, err end
    
    ok, err = manif.update_manifest(name, version)