diff --git a/src/Components/Document/DocEditPage.js b/src/Components/Document/DocEditPage.js
index 936f001..fc53b1e 100644
--- a/src/Components/Document/DocEditPage.js
+++ b/src/Components/Document/DocEditPage.js
@@ -47,7 +47,8 @@ const DocEditPage = () => {
   const [index, setIndex] = useState(0);
 
   // For document generation:
-  const state = group ? group?.state : null;
+
+  const state = group.state[0] ? group.state[0].code : "ny";
   const lawFirm = group ? group?.firm : null;
   const streetAdd = group ? group?.streetAddress : null;
   const city = group ? group.city : null;
@@ -61,14 +62,14 @@ const DocEditPage = () => {
       : process.env.REACT_APP_API_PROD;
 
   const displayCopy =
-    state === "New York"
+    state === "ny"
       ? docEditCopy.NewYork
-      : state === "New Jersey"
+      : state === "nj"
       ? docEditCopy.NewJersey
       : docEditCopy.NewYork;
 
   const headerPicker = () => {
-    switch (state.code) {
+    switch (state) {
       case "ny":
         return (
           <NewYorkCaption