diff --git a/app/src/main/java/dev/kaizen/app/chat/ChatScreen.kt b/app/src/main/java/dev/kaizen/app/chat/ChatScreen.kt
index 0855a90..bf847cf 100644
--- a/app/src/main/java/dev/kaizen/app/chat/ChatScreen.kt
+++ b/app/src/main/java/dev/kaizen/app/chat/ChatScreen.kt
@@ -1033,9 +1033,6 @@ fun ChatScreen(
                                     }
                                 }
                                 Spacer(Modifier.weight(1f))
-                                val ctxLen = models.find { it.id == displayModel }?.context_length ?: 1_000_000
-                                TokenBadge(used = usedTokens, contextLength = ctxLen)
-                                Spacer(Modifier.width(8.dp))
                                 Box(
                                     Modifier.size(36.dp).clip(KaizenShapes.circle)
                                         .clickable { haptics.tick(); newChat() },
@@ -1047,6 +1044,9 @@ fun ChatScreen(
                                         modifier = Modifier.size(17.dp),
                                     )
                                 }
+                                Spacer(Modifier.width(4.dp))
+                                val ctxLen = models.find { it.id == displayModel }?.context_length ?: 1_000_000
+                                TokenBadge(used = usedTokens, contextLength = ctxLen)
                             }
                         }