diff --git a/components/WeaponUnit/index.tsx b/components/WeaponUnit/index.tsx
index 21df410e..f90067af 100644
--- a/components/WeaponUnit/index.tsx
+++ b/components/WeaponUnit/index.tsx
@@ -79,15 +79,13 @@ const WeaponUnit = (props: Props) => {
     )
 
     const editableImage = (
-        <div className="WeaponImage">
-            <SearchModal 
-                placeholderText="Search for a weapon..." 
-                fromPosition={props.position} 
-                object="weapons"
-                send={props.updateObject}>
-                    {image}
-            </SearchModal>
-        </div>
+        <SearchModal 
+            placeholderText="Search for a weapon..." 
+            fromPosition={props.position} 
+            object="weapons"
+            send={props.updateObject}>
+                {image}
+        </SearchModal>
     )
 
     return (