1 files changed, 5 insertions, 9 deletions
diff --git a/jb-ui/src/JBApp.tsx b/jb-ui/src/JBApp.tsx
index fd0eeba..a69cc85 100644
--- a/jb-ui/src/JBApp.tsx
+++ b/jb-ui/src/JBApp.tsx
@@ -16,6 +16,7 @@ import {
 import Footer from "@/components/Footer";
 import { useAppDispatch } from "@/hooks";
 import { bpid, routeBasename, tagManager } from "@/lib/utils";
+import { useGetMeQuery } from "@/models/authAPI";
 import {
   setAssignmentID,
   setProductUserID,
@@ -72,24 +73,19 @@ function isPingMessage(msg: Message): msg is PingMessage {
 function QueryParamProcessor() {
   const dispatch = useAppDispatch();
   const [searchParams] = useSearchParams();
+  const { data: user, isLoading, isError } = useGetMeQuery();
 
   useEffect(() => {
-    const worker_id = searchParams.get("workerId");
     const assignment_id = searchParams.get("assignmentId");
     const tsid = searchParams.get("tsid");
-    const turkSubmitTo = searchParams.get("turkSubmitTo");
 
-    if (turkSubmitTo) {
-      dispatch(setTurkSubmitTo(turkSubmitTo));
-    }
-
-    if (worker_id) {
-      dispatch(setProductUserID(worker_id));
+    if (user?.product_user_id) {
+      dispatch(setProductUserID(user.product_user_id));
 
       new WalletApi()
         .getUserWalletBalanceProductIdWalletGet(
           bpid, // productId
-          worker_id, // bpuid
+          user.product_user_id,
         )
         .then((res) => {
           const response = res.data as UserWalletBalanceResponse;