Error when removing a non-existing file.. duh..

1 files changed, 4 insertions, 2 deletions

diff --git a/archinstall.py b/archinstall.py
index 26b1757e..020f9790 100644
--- a/archinstall.py
+++ b/archinstall.py
@@ -200,8 +200,10 @@ def update_git():
 	if(default_gw):
 		print('[N] Updating git!')
 		## Not the most elegant way to make sure git conflicts doesn't occur (yea fml)
-		os.remove('/root/archinstall/archinstall.py')
-		os.remove('/root/archinstall/README.md')
+		if os.path.isfile('/root/archinstall/archinstall.py'):
+			os.remove('/root/archinstall/archinstall.py')
+		if os.path.isfile('/root/archinstall/README.md'):
+			os.remove('/root/archinstall/README.md')
 
 		simple_command('(cd /root/archinstall; git reset --hard origin/$(git branch | grep "*" | cut -d\' \' -f 2))') # git reset --hard origin/<branch_name> / git fetch --all
 		print(output)