diff --git a/literalai/api/gql.py b/literalai/api/gql.py
index 7ad4713..1aab01e 100644
--- a/literalai/api/gql.py
+++ b/literalai/api/gql.py
@@ -1024,17 +1024,20 @@
 GET_PROMPT_VERSION = """
 query GetPrompt($id: String, $name: String, $version: Int) {
     promptVersion(id: $id, name: $name, version: $version) {
-        id
         createdAt
-        updatedAt
-        type
+        id
+        label
+        settings
+        status
+        tags
         templateMessages
         tools
-        settings
+        type
+        updatedAt
+        url
         variables
         variablesDefaultValues
         version
-        url
         lineage {
             name
         }
diff --git a/tests/e2e/test_e2e.py b/tests/e2e/test_e2e.py
index 992c3a5..aede802 100644
--- a/tests/e2e/test_e2e.py
+++ b/tests/e2e/test_e2e.py
@@ -176,6 +176,7 @@ async def test_step(self, client: LiteralClient, async_client: AsyncLiteralClien
         await async_client.flush()
         is_deleted = await async_client.api.delete_step(id=step.id)
         assert is_deleted is True, "Step should be deleted"
+        time.sleep(1)
 
         deleted_step = await async_client.api.get_step(id=step.id)