diff --git a/apps/wallet/views/application.py b/apps/wallet/views/application.py
index 74821b6..4f75456 100755
--- a/apps/wallet/views/application.py
+++ b/apps/wallet/views/application.py
@@ -9,7 +9,7 @@ from apps.users.models import User
 from apps.wallet.constans import TypeChoices, StateChoices
 from apps.wallet.models import Transaction
 from apps.wallet.serializers import ApplicationDepositSerializer, ApplicationWithdrawSerializer, \
-    ApplicationTransactionSerializer, ApplicationBalanceSerializer
+    ApplicationTransactionSerializer, ApplicationBalanceSerializer, MyWalletSerializer
 
 
 def get_application(request):
@@ -37,12 +37,12 @@ class UserBalanceAPIView(generics.RetrieveAPIView):
     API for getting specified user or application balance.
     '''
     permission_classes = [IsAuthenticatedOrTokenHasScope]
-    serializer_class = ApplicationBalanceSerializer
+    serializer_class = MyWalletSerializer
     required_scopes = ['wallet.application:get_user_balance']
 
     def get_object(self):
         user_uuid = self.kwargs.get('uuid')
-        user = User.objects.get_or_create(uuid=user_uuid)
+        user, created = User.objects.get_or_create(uuid=user_uuid)
         # user = get_object_or_404(User.objects.all(), uuid=user_uuid)
 
         return user