diff --git a/admin_api/views.py b/admin_api/views.py
index 5bd4145..c5e846c 100644
--- a/admin_api/views.py
+++ b/admin_api/views.py
@@ -2610,6 +2610,7 @@ class PartnerImpersonateView(APIView):
 
     def post(self, request, pk):
         from partner.models import Partner as PartnerModel
+        from django.shortcuts import get_object_or_404
         partner = get_object_or_404(PartnerModel, pk=pk)
         partner_user = User.objects.filter(partner=partner, role='partner_manager').first()
         if not partner_user: