diff --git a/server/src/browser-management/classes/RemoteBrowser.ts b/server/src/browser-management/classes/RemoteBrowser.ts
index 305f4a62..60c43199 100644
--- a/server/src/browser-management/classes/RemoteBrowser.ts
+++ b/server/src/browser-management/classes/RemoteBrowser.ts
@@ -153,6 +153,8 @@ export class RemoteBrowser {
         this.socket.on('setViewportSize', async (data: { width: number, height: number }) => {
             const { width, height } = data;
             logger.log('debug', `Received viewport size: width=${width}, height=${height}`);
+
+            this.viewportSize = { width, height };
     
             // Update the browser context's viewport dynamically
             if (this.context && this.browser) {