diff --git a/push_notifications/api/rest_framework.py b/push_notifications/api/rest_framework.py
index 6ad4a546..4f4e8d10 100644
--- a/push_notifications/api/rest_framework.py
+++ b/push_notifications/api/rest_framework.py
@@ -138,9 +138,9 @@ class DeviceViewSetMixin(object):
 	def create(self, request, *args, **kwargs):
 		serializer = None
 		is_update = False
-		if SETTINGS.get("UPDATE_ON_DUPLICATE_REG_ID") and "registration_id" in request.data:
+		if SETTINGS.get("UPDATE_ON_DUPLICATE_REG_ID") and self.lookup_field in request.data:
 			instance = self.queryset.model.objects.filter(
-				registration_id=request.data["registration_id"]
+				registration_id=request.data[self.lookup_field]
 			).first()
 			if instance:
 				serializer = self.get_serializer(instance, data=request.data)