diff --git a/src/one-shot/schema.py b/src/one-shot/schema.py
index 6ea2676df42d246faeaee0a96794802844d1b2d6..d33f7728960a42b889a6a1c66515684ad6c482ba 100644
--- a/src/one-shot/schema.py
+++ b/src/one-shot/schema.py
@@ -5,29 +5,6 @@ from pydantic import BaseModel
 
 
 class MultiWOZ(BaseModel):
-    attractionArea: Optional[Literal[
-        "center",
-        "east",
-        "north",
-        "south",
-        "west",
-    ]]
-    attractionName: Optional[str]
-    attractionType: Optional[Literal[
-        "architecture",
-        "boat",
-        "cinema",
-        "college",
-        "college",
-        "concerthall"
-        "entertainment",
-        "multiple sports",
-        "museum",
-        "nightclub",
-        "park",
-        "swimmingpool",
-        "theatre",
-    ]]
     hotelArea: Optional[Literal[
         "center",
         "east",
@@ -83,6 +60,29 @@ class MultiWOZ(BaseModel):
     trainDeparture: Optional[datetime]
     trainDestination: Optional[str]
     trainLeaveat: Optional[datetime]
+    attractionArea: Optional[Literal[
+        "center",
+        "east",
+        "north",
+        "south",
+        "west",
+    ]]
+    attractionName: Optional[str]
+    attractionType: Optional[Literal[
+        "architecture",
+        "boat",
+        "cinema",
+        "college",
+        "college",
+        "concerthall"
+        "entertainment",
+        "multiple sports",
+        "museum",
+        "nightclub",
+        "park",
+        "swimmingpool",
+        "theatre",
+    ]]
 
 
 class HeidelKBerg(BaseModel):