diff --git a/archinstall.py b/archinstall.py
index bddf8de4..61a51e36 100644
--- a/archinstall.py
+++ b/archinstall.py
@@ -789,7 +789,7 @@ def cache_diskpw_on_disk():
 		with open(args['pwfile'], 'w') as pw:
 			pw.write(args['password'])
 
-def refresh_partition_list(drive):
+def refresh_partition_list(drive, *args, **kwargs):
 	args['paritions'] = get_partitions(drive)
 
 if __name__ == '__main__':