diff --git a/packages/Webkul/API/Http/Controllers/Shop/SessionController.php b/packages/Webkul/API/Http/Controllers/Shop/SessionController.php
index 19370d917..b84f3694d 100644
--- a/packages/Webkul/API/Http/Controllers/Shop/SessionController.php
+++ b/packages/Webkul/API/Http/Controllers/Shop/SessionController.php
@@ -103,23 +103,19 @@ class SessionController extends Controller
             'password'      => 'confirmed|min:6',
         ]);
 
-        $data = request()->all();
+        $data = request()->only('first_name', 'last_name', 'gender', 'date_of_birth', 'email', 'password');
 
-        if (! $data['date_of_birth']) {
-            unset($data['date_of_birth']);
-        }
-
-        if (!isset($data['password']) || ! $data['password']) {
+        if (! isset($data['password']) || ! $data['password']) {
             unset($data['password']);
         } else {
             $data['password'] = bcrypt($data['password']);
         }
 
-        $this->customerRepository->update($data, $customer->id);
+        $updatedCustomer = $this->customerRepository->update($data, $customer->id);
 
         return response()->json([
-            'message' => 'Your account has been created successfully.',
-            'data'    => new CustomerResource($this->customerRepository->find($customer->id)),
+            'message' => 'Your account has been updated successfully.',
+            'data'    => new CustomerResource($updatedCustomer),
         ]);
     }
 
@@ -136,4 +132,4 @@ class SessionController extends Controller
             'message' => 'Logged out successfully.',
         ]);
     }
-}
\ No newline at end of file
+}