diff --git a/main/main.cpp b/main/main.cpp
index 49a775d..a13a01d 100644
--- a/main/main.cpp
+++ b/main/main.cpp
@@ -23,8 +23,8 @@ int main() {
     std::string currentText;
 
     /*************************set back to shared memory(only for test)********************************************/
-    int windowWidth = 40;
-    int windowHeight = 15;
+    int windowWidth = vkvm::getWidth();
+    int windowHeight = vkvm::getHeight();
     vkvm::setWidth(windowWidth);
     vkvm::setHeight(windowHeight);