diff --git a/taiga/base/api.py b/taiga/base/api.py
index 87a84b20..c8630df2 100644
--- a/taiga/base/api.py
+++ b/taiga/base/api.py
@@ -146,7 +146,8 @@ class ReversionMixin(object):
         return Response(serializer.data)
 
     @rf_decorators.action()
-    def restore(self, request, vpk=None, *args, **kwargs):
+    def restore(self, request, *args, **kwargs):
+        vpk = request.QUERY_PARAMS.get("version", None)
         if not vpk:
             return Response(status=status.HTTP_404_NOT_FOUND)