diff --git a/src/ogRest.py b/src/ogRest.py
index 2bae2ab..c0b886d 100644
--- a/src/ogRest.py
+++ b/src/ogRest.py
@@ -142,7 +142,7 @@ class ogThread():
 		client.send(restResponse.getResponse(ogResponses.OK, jsonResp))
 
 	# Process image restore
-	def procirestore(httpparser, ogRest):
+	def procirestore(client, httpparser, ogRest):
 		try:
 			ogOperations.procirestore(httpparser, ogRest)
 		except ValueError as err: