diff --git a/src/state/mod.rs b/src/state/mod.rs
index 5cf99f8..84387ca 100644
--- a/src/state/mod.rs
+++ b/src/state/mod.rs
@@ -106,30 +106,35 @@ impl State {
 }
 
 
+#[cfg(test)]
+mod tests {
+    use crate::state::State;
+    use crate::state::types::ElementContent;
+    
+    #[tokio::test]
+    #[serial_test::serial]
+    async fn test_create() {
+        tracing_subscriber::fmt().pretty().init();
+        let state = State::new().await.unwrap();
+        let id = state.create_element(&ElementContent::Text("Test-text".to_string())).unwrap();
+        let el = state.get_element(&id).unwrap();
+        assert_eq!(
+            ElementContent::Text("Test-text".to_string()),
+            el.content().to_owned()
+        )
+    }
 
-#[tokio::test]
-#[serial_test::serial]
-async fn test_create() {
-    tracing_subscriber::fmt().pretty().init();
-    let state = State::new().await.unwrap();
-    let id = state.create_element(&ElementContent::Text("Test-text".to_string())).unwrap();
-    let el = state.get_element(&id).unwrap();
-    assert_eq!(
-        ElementContent::Text("Test-text".to_string()),
-        el.content().to_owned()
-    )
+    #[tokio::test]
+    #[serial_test::serial]
+    async fn test_update() {
+        tracing_subscriber::fmt().pretty().init();
+        let state = State::new().await.unwrap();
+        let id = state.create_element(&ElementContent::Text("Test-text".to_string())).unwrap();
+        state.set_element(&id,&ElementContent::Text("Test-text 2".to_string())).unwrap();
+        let el = state.get_element(&id).unwrap();
+        assert_eq!(
+            ElementContent::Text("Test-text 2".to_string()),
+            el.content().to_owned()
+        )
+    }
 }
-
-#[tokio::test]
-#[serial_test::serial]
-async fn test_update() {
-    tracing_subscriber::fmt().pretty().init();
-    let state = State::new().await.unwrap();
-    let id = state.create_element(&ElementContent::Text("Test-text".to_string())).unwrap();
-    state.set_element(&id,&ElementContent::Text("Test-text 2".to_string())).unwrap();
-    let el = state.get_element(&id).unwrap();
-    assert_eq!(
-        ElementContent::Text("Test-text 2".to_string()),
-        el.content().to_owned()
-    )
-}
\ No newline at end of file
diff --git a/src/state/types/element_content.rs b/src/state/types/element_content.rs
index b14f7cc..8655273 100644
--- a/src/state/types/element_content.rs
+++ b/src/state/types/element_content.rs
@@ -26,9 +26,4 @@ impl TryFrom<&str> for ElementContent {
             Err(e) => Err(e)
         }
     }
-}
-
-#[test]
-pub fn test() {
-    
 }
\ No newline at end of file