diff --git a/src/ui/src/components/Dashboard.vue b/src/ui/src/components/Dashboard.vue
index 3920d2d5..465a436d 100644
--- a/src/ui/src/components/Dashboard.vue
+++ b/src/ui/src/components/Dashboard.vue
@@ -80,7 +80,7 @@
       </v-window-item>
       <v-window-item :key="3" :value="3">
         <v-row class="pa-1 ma-1 justify-center">
-          <Experiments :experiment_objects="experiment_objects" :wc_state="wc_state" :campaigns="campaigns"/>
+          <Experiments/>
         </v-row>
       </v-window-item>
     </v-window>
@@ -88,7 +88,9 @@
 </template>
 
 <script setup lang="ts">
-import { ref, watchEffect } from 'vue';
+/// <reference path="../store.d.ts" />
+import { main_url, wc_info, wc_state, wfs } from "@/store";
+import { ref } from 'vue';
 import VueJsonPretty from 'vue-json-pretty';
 import 'vue-json-pretty/lib/styles.css';
 import CancelButton from './AdminButtons/CancelButton.vue';
@@ -97,75 +99,7 @@ import PauseResumeButton from './AdminButtons/PauseResumeButton.vue';
 import ResetButton from './AdminButtons/ResetButton.vue';
 import SafetyStopButton from './AdminButtons/SafetyStopButton.vue';
 import ShutdownButton from './AdminButtons/ShutdownButton.vue';
-const main_url = ref()
-const state_url = ref()
-const workcell_info_url = ref()
-const has_url = ref(false)
 const tab = ref(1)
-const wfs = ref([''])
-const experiments = ref()
-const experiments_url = ref()
-const wc_state = ref()
-const wc_info = ref()
-const campaigns = ref()
-const campaigns_url = ref()
-const experiment_keys = ref()
-const experiment_objects: any = ref([])
-main_url.value = "http://".concat(window.location.host)
-class ExperimentInfo {
-  experiment_id?: string;
-  experiment_workflows: any;
-  experiment_name?: string;
-  num_wfs?: any;
-  num_events?: any;
-  events?: any
-}
-async function get_events(experiment_id: string) {
-  return Object.values(await ((await fetch(main_url.value.concat("/experiments/".concat(experiment_id).concat("/events"))))).json());
-}
-watchEffect(async () => {
-  has_url.value = true;
-  state_url.value = main_url.value.concat("/wc/state")
-
-  experiments_url.value = main_url.value.concat("/experiments/all")
-  campaigns_url.value = experiments_url.value.concat("/campaigns/all")
-  workcell_info_url.value = main_url.value.concat("/wc/")
-
-  watchEffect(async () => wc_state.value = await (await fetch(state_url.value)).json())
-  watchEffect(async () => wc_info.value = await (await fetch(workcell_info_url.value)).json())
-
-  // var new_experiment_keys = [];
-  // experiment_keys.value = [];
-  setInterval(updateDashboard, 1000)
-  setInterval(updateExperiments, 10000)
-
-  async function updateExperiments() {
-    experiment_objects.value = Object.values(experiments.value);
-    campaigns.value = await ((await fetch(campaigns_url.value)).json());
-    // new_experiment_keys = Object.keys(experiments.value).sort();
-    // let difference = new_experiment_keys.filter(x => !experiment_keys.value.includes(x));
-    // experiment_keys.value = Object.keys(experiments.value).sort();
-    // difference.forEach(async function (value: any) {
-    //   var experiment: ExperimentInfo = new ExperimentInfo();
-    //   experiment.experiment_id = value;
-    //   var events = await get_events(value);
-
-    //   experiment.experiment_name = experiments.value[value].experiment_name;
-    //   experiment.experiment_workflows = wfs.value.filter((key: any) => wc_state.value.workflows[key].experiment_id === value);
-    //   experiment.events = events;
-    //   experiment.num_wfs = experiment.experiment_workflows.length;
-    //   experiment.num_events = experiment.events.length;
-    //   experiment_objects.value.splice(0, 0, experiment);
-    // });
-  }
-
-  async function updateDashboard() {
-    wc_state.value = await (await fetch(state_url.value)).json();
-    wfs.value = Object.keys(wc_state.value.workflows).sort().reverse();
-    experiments.value = await ((await fetch(experiments_url.value)).json());
-  }
-}
-)
 </script>
 
 <script lang="ts">
diff --git a/src/ui/src/components/Event.vue b/src/ui/src/components/Event.vue
index a656c6e5..c10991f8 100644
--- a/src/ui/src/components/Event.vue
+++ b/src/ui/src/components/Event.vue
@@ -1,11 +1,13 @@
 <template>
-    <vue-json-pretty :data="event" />
+    <vue-json-pretty :data="events[props.event_id]" />
 </template>
 
 <script setup lang="ts">
 
 import VueJsonPretty from 'vue-json-pretty';
 import 'vue-json-pretty/lib/styles.css';
-const props = defineProps(['event', "wc_state"])
+/// <reference path="../store.d.ts" />
+import { events } from "@/store";
+const props = defineProps(['event_id'])
 
 </script>
diff --git a/src/ui/src/components/Experiments.vue b/src/ui/src/components/Experiments.vue
index f07d2a08..a2746d70 100644
--- a/src/ui/src/components/Experiments.vue
+++ b/src/ui/src/components/Experiments.vue
@@ -7,7 +7,7 @@
       </v-toolbar>
     </template>
     <template v-slot:item.campaign_id="{ value }">
-      <td>{{ (value != null && value in campaigns) ? campaigns[value].campaign_name : "-" }}</td>
+      <td>{{ (value != null && campaigns !== undefined && value in campaigns) ? campaigns[value].campaign_name : "-" }}</td>
     </template>
     <!-- <template v-slot:expanded-row="{ columns, item}: {columns: any, item: any}">
       <tr>
@@ -31,7 +31,8 @@
 
 <script setup lang="ts">
 import { VDataTable } from 'vuetify/lib/components/index.mjs';
-const props = defineProps(["experiment_objects", "wc_state", "campaigns"])
+/// <reference path="../store.d.ts" />
+import { campaigns, experiment_objects } from "@/store";
 const sortBy: VDataTable['sortBy'] = [{ key: 'experiment_id', order:'desc'}];
 
 const arg_headers = [
diff --git a/src/ui/src/components/ModuleModal.vue b/src/ui/src/components/ModuleModal.vue
index 40cc3276..8fb9eb17 100644
--- a/src/ui/src/components/ModuleModal.vue
+++ b/src/ui/src/components/ModuleModal.vue
@@ -1,5 +1,5 @@
 <template>
-  <v-dialog class="pa-3" v-slot:default="{ isActive }" max-width="1000">
+  <v-dialog class="pa-3" v-slot:default="{ isActive }" max-width="1920">
     <v-card>
       <v-card-title>
         <div class="d-flex align-center w-100">
@@ -148,6 +148,7 @@ const arg_headers = [
   { title: 'Required', key: 'required' },
   { title: 'Default', key: 'default' },
   { title: 'Description', key: 'description' },
+  { title: "Value", minWidth: "200px"}
 ]
 const copy = ref(false)
 const file_headers = [
diff --git a/src/ui/src/store.d.ts b/src/ui/src/store.d.ts
new file mode 100644
index 00000000..1cb4523e
--- /dev/null
+++ b/src/ui/src/store.d.ts
@@ -0,0 +1,18 @@
+
+export { };
+
+declare module "store" {
+
+  export interface Store{
+    campaigns: any;
+    campaigns_url: string;
+    experiment_keys: any;
+    experiment_objects: any;
+    experiments: any;
+    experiments_url: string;
+    main_url: string;
+    state_url: string;
+    wc_info: any;
+    wc_state: any;
+  }
+}
diff --git a/src/ui/src/store.ts b/src/ui/src/store.ts
new file mode 100644
index 00000000..dcc37b37
--- /dev/null
+++ b/src/ui/src/store.ts
@@ -0,0 +1,67 @@
+
+import { ref, watchEffect } from 'vue';
+
+const main_url = ref()
+const state_url = ref()
+const workcell_info_url = ref()
+const wfs = ref([''])
+const experiments = ref()
+const experiments_url = ref()
+const events_url = ref()
+const events = ref()
+const wc_state = ref()
+const wc_info = ref()
+const campaigns = ref()
+const campaigns_url = ref()
+const experiment_keys = ref()
+const experiment_objects: any = ref([])
+main_url.value = "http://".concat(window.location.host)
+class ExperimentInfo {
+    experiment_id?: string;
+    experiment_workflows: any;
+    experiment_name?: string;
+    num_wfs?: any;
+    num_events?: any;
+    events?: any
+}
+async function get_events(experiment_id: string) {
+    return Object.values(await ((await fetch(main_url.value.concat("/experiments/".concat(experiment_id).concat("/events"))))).json());
+}
+watchEffect(async () => {
+    state_url.value = main_url.value.concat("/wc/state")
+
+    experiments_url.value = main_url.value.concat("/experiments/all")
+    campaigns_url.value = main_url.value.concat("/campaigns/all")
+    workcell_info_url.value = main_url.value.concat("/wc/")
+    events_url.value = main_url.value.concat("/events/all")
+
+    setInterval(updateWorkcellState, 1000)
+    setInterval(updateWorkflows, 1000)
+    setInterval(updateExperiments, 10000)
+    setInterval(updateCampaigns, 10000);
+    setInterval(updateEvents, 10000);
+
+    async function updateExperiments() {
+        experiments.value = await ((await fetch(experiments_url.value)).json());
+        experiment_objects.value = Object.values(experiments.value);
+
+    }
+
+    async function updateCampaigns() {
+        campaigns.value = await ((await fetch(campaigns_url.value)).json());
+    }
+
+    async function updateEvents() {
+        events.value = await ((await fetch(events_url.value)).json());
+    }
+
+    async function updateWorkcellState() {
+        wc_state.value = await (await fetch(state_url.value)).json();
+    }
+
+    async function updateWorkflows() {
+        wfs.value = Object.keys(wc_state.value.workflows).sort().reverse();
+    }
+})
+
+export { campaigns, campaigns_url, events, experiment_keys, experiment_objects, experiments, experiments_url, main_url, state_url, wc_info, wc_state, wfs, workcell_info_url };
diff --git a/src/ui/vite.config.mts b/src/ui/vite.config.mts
index b8d760ef..13fcbbff 100644
--- a/src/ui/vite.config.mts
+++ b/src/ui/vite.config.mts
@@ -1,12 +1,12 @@
 // Plugins
-import Components from 'unplugin-vue-components/vite'
 import Vue from '@vitejs/plugin-vue'
-import Vuetify, { transformAssetUrls } from 'vite-plugin-vuetify'
 import ViteFonts from 'unplugin-fonts/vite'
+import Components from 'unplugin-vue-components/vite'
+import Vuetify, { transformAssetUrls } from 'vite-plugin-vuetify'
 
 // Utilities
-import { defineConfig } from 'vite'
 import { fileURLToPath, URL } from 'node:url'
+import { defineConfig } from 'vite'
 
 // https://vitejs.dev/config/
 export default defineConfig({
@@ -37,6 +37,7 @@ export default defineConfig({
       '.jsx',
       '.mjs',
       '.ts',
+      '.d.ts',
       '.tsx',
       '.vue',
     ],
diff --git a/src/wei/core/state_manager.py b/src/wei/core/state_manager.py
index 93f42af7..d6d39ca5 100644
--- a/src/wei/core/state_manager.py
+++ b/src/wei/core/state_manager.py
@@ -7,6 +7,7 @@
 
 import redis
 from pottery import InefficientAccessWarning, RedisDict, Redlock
+from pydantic import ValidationError
 
 from wei.config import Config
 from wei.types import Module, Workcell, WorkflowRun
@@ -135,7 +136,7 @@ def shutdown(self, value: bool) -> None:
             f"{self._workcell_prefix}:shutdown", "true" if value else "false"
         )
 
-    # Locking Methods
+    # *Locking Methods
     def campaign_lock(self, campaign_id: str) -> Redlock:
         """
         Get a lock on a particular campaign. This should be called before editing a Campaign.
@@ -167,7 +168,7 @@ def wc_state_lock(self) -> Redlock:
             auto_release_time=60,
         )
 
-    # State Methods
+    # *State Methods
     def get_state(self) -> Dict[str, Dict[Any, Any]]:
         """
         Return a dict containing the current state of the workcell.
@@ -241,21 +242,24 @@ def has_state_changed(self) -> bool:
         else:
             return False
 
-    # Campaign Methods
+    # *Campaign Methods
     def get_campaign(self, campaign_id: str) -> Campaign:
         """
         Returns a campaign by ID
         """
-        return self._campaigns[campaign_id]
+        return Campaign.model_validate(self._campaigns[campaign_id])
 
     def get_all_campaigns(self) -> Dict[str, Campaign]:
         """
         Returns all campaigns
         """
-        return {
-            str(campaign_id): Campaign.model_validate(campaign)
-            for campaign_id, campaign in self._campaigns.to_dict().items()
-        }
+        valid_campaigns = {}
+        for campaign_id, campaign in self._campaigns.to_dict().items():
+            try:
+                valid_campaigns[str(campaign_id)] = Campaign.model_validate(campaign)
+            except ValidationError:
+                continue
+        return valid_campaigns
 
     def set_campaign(self, campaign: Campaign) -> None:
         """
@@ -277,7 +281,7 @@ def update_campaign(
         """
         self.set_campaign(func(self.get_campaign(campaign_id), *args, **kwargs))
 
-    # Experiment Methods
+    # *Experiment Methods
     def get_experiment(self, experiment_id: str) -> Experiment:
         """
         Returns an experiment by ID
@@ -288,12 +292,15 @@ def get_all_experiments(self) -> Dict[str, Experiment]:
         """
         Returns all experiments
         """
-        return {
-            str(experiment_id): Experiment.model_validate(
-                experiment, from_attributes=True
-            )
-            for experiment_id, experiment in self._experiments.to_dict().items()
-        }
+        valid_experiments = {}
+        for experiment_id, experiment in self._experiments.to_dict().items():
+            try:
+                valid_experiments[str(experiment_id)] = Experiment.model_validate(
+                    experiment, from_attributes=True
+                )
+            except ValidationError:
+                continue
+        return valid_experiments
 
     def set_experiment(self, experiment: Experiment) -> None:
         """
@@ -315,7 +322,7 @@ def update_experiment(
         """
         self.set_experiment(func(self.get_experiment(experiment_id), *args, **kwargs))
 
-    # Event Methods
+    # *Event Methods
     def get_event(self, event_id: str) -> Event:
         """
         Returns an event by ID
@@ -326,10 +333,15 @@ def get_all_events(self) -> Dict[str, Event]:
         """
         Returns all events
         """
-        return {
-            str(event_id): Event.model_validate(event)
-            for event_id, event in self._events.to_dict().items()
-        }
+        from pydantic import ValidationError
+
+        valid_events = {}
+        for event_id, event in self._events.to_dict().items():
+            try:
+                valid_events[str(event_id)] = Event.model_validate(event)
+            except ValidationError:
+                continue
+        return valid_events
 
     def set_event(self, event: Event) -> None:
         """
@@ -337,7 +349,7 @@ def set_event(self, event: Event) -> None:
         """
         self._events[event.event_id] = event.model_dump(mode="json")
 
-    # DataPoint Methods
+    # *DataPoint Methods
     def get_datapoint(self, data_id: str) -> DataPoint:
         """
         Returns an event by ID
@@ -348,12 +360,17 @@ def get_datapoint(self, data_id: str) -> DataPoint:
 
     def get_all_datapoints(self) -> Dict[str, DataPoint]:
         """
-        Returns all events
+        Returns all datapoints
         """
-        return {
-            str(datapoint_id): DataPoint.model_validate(datapoint)
-            for datapoint_id, datapoint in self._datapoints.to_dict().items()
-        }
+        valid_datapoints = {}
+        for datapoint_id, datapoint in self._datapoints.to_dict().items():
+            try:
+                valid_datapoints[str(datapoint_id)] = DataPoint.model_validate(
+                    datapoint
+                )
+            except ValidationError:
+                continue
+        return valid_datapoints
 
     def set_datapoint(self, datapoint: DataPoint) -> None:
         """
@@ -361,7 +378,7 @@ def set_datapoint(self, datapoint: DataPoint) -> None:
         """
         self._datapoints[datapoint.id] = datapoint.model_dump(mode="json")
 
-    # Workcell Methods
+    # *Workcell Methods
     def get_workcell(self) -> Workcell:
         """
         Returns the current workcell as a Workcell object
@@ -392,7 +409,7 @@ def get_workcell_id(self) -> str:
             wc_id = self._redis_client.get(f"{self._workcell_prefix}:workcell_id")
         return wc_id
 
-    # Workflow Methods
+    # *Workflow Methods
     def get_workflow_run(self, run_id: Union[str, str]) -> WorkflowRun:
         """
         Returns a workflow by ID
@@ -401,12 +418,17 @@ def get_workflow_run(self, run_id: Union[str, str]) -> WorkflowRun:
 
     def get_all_workflow_runs(self) -> Dict[str, WorkflowRun]:
         """
-        Returns all workflows
+        Returns all workflow runs
         """
-        return {
-            str(run_id): WorkflowRun.model_validate(workflow_run)
-            for run_id, workflow_run in self._workflow_runs.to_dict().items()
-        }
+        valid_workflow_runs = {}
+        for run_id, workflow_run in self._workflow_runs.to_dict().items():
+            try:
+                valid_workflow_runs[str(run_id)] = WorkflowRun.model_validate(
+                    workflow_run
+                )
+            except ValidationError:
+                continue
+        return valid_workflow_runs
 
     def set_workflow_run(self, wf: WorkflowRun) -> None:
         """
@@ -434,7 +456,7 @@ def update_workflow_run(
         """
         self.set_workflow_run(func(self.get_workflow_run(run_id), *args, **kwargs))
 
-    # Location Methods
+    # *Location Methods
     def get_location(self, location_name: str) -> Location:
         """
         Returns a location by name
@@ -445,10 +467,13 @@ def get_all_locations(self) -> Dict[str, Location]:
         """
         Returns all locations
         """
-        return {
-            str(location_name): Location.model_validate(location)
-            for location_name, location in self._locations.to_dict().items()
-        }
+        valid_locations = {}
+        for location_name, location in self._locations.to_dict().items():
+            try:
+                valid_locations[str(location_name)] = Location.model_validate(location)
+            except ValidationError:
+                continue
+        return valid_locations
 
     def set_location(
         self, location_name: str, location: Union[Location, Dict[str, Any]]
@@ -480,7 +505,7 @@ def update_location(
             location_name, func(self.get_location(location_name), *args, **kwargs)
         )
 
-    # Module Methods
+    # *Module Methods
     def get_module(self, module_name: str) -> Module:
         """
         Returns a module by name
@@ -489,12 +514,15 @@ def get_module(self, module_name: str) -> Module:
 
     def get_all_modules(self) -> Dict[str, Module]:
         """
-        Returns a module by name
+        Returns all modules
         """
-        return {
-            str(module_name): Module.model_validate(module)
-            for module_name, module in self._modules.to_dict().items()
-        }
+        valid_modules = {}
+        for module_name, module in self._modules.to_dict().items():
+            try:
+                valid_modules[str(module_name)] = Module.model_validate(module)
+            except ValidationError:
+                continue
+        return valid_modules
 
     def set_module(
         self, module_name: str, module: Union[Module, ModuleDefinition, Dict[str, Any]]
diff --git a/src/wei/experiment_client.py b/src/wei/experiment_client.py
index 5e76b7ba..c3d236a8 100644
--- a/src/wei/experiment_client.py
+++ b/src/wei/experiment_client.py
@@ -53,8 +53,8 @@ def __init__(
         self,
         server_host: str = "localhost",
         server_port: str = "8000",
-        experiment: Optional[Union[ExperimentDesign, Experiment, ULID]] = None,
-        campaign: Optional[Union[CampaignDesign, Campaign, ULID]] = None,
+        experiment: Optional[Union[ExperimentDesign, Experiment, str]] = None,
+        campaign: Optional[Union[CampaignDesign, Campaign, str]] = None,
         working_dir: Optional[PathLike] = None,
         log_experiment_end_on_exit: bool = True,
         timeout=10,
@@ -69,10 +69,10 @@ def __init__(
         server_port: str
             port for WEI server
 
-        experiment: Optional[Union[ExperimentDesign, Experiment, ULID]]
+        experiment: Optional[Union[ExperimentDesign, Experiment, str]]
             A new experiment design, or an existing experiment or experiment_id to continue
 
-        campaign: Optional[Union[CampaignDesign, Campaign, ULID]]
+        campaign: Optional[Union[CampaignDesign, Campaign, str]]
             A new campaign design, or an existing campaign or campaign_id to associate with the experiment
 
         working_dir: Optional[Union[str, Path]]
@@ -89,6 +89,7 @@ def __init__(
         self.timeout = timeout
         self.experiment = experiment
         self.campaign = campaign
+        self.working_dir = Path(working_dir) or Path.cwd()
 
         if experiment:
             self.start_or_continue_experiment(experiment, campaign)
@@ -133,17 +134,17 @@ def _connect_to_server(self):
 
     def start_or_continue_experiment(
         self,
-        experiment: Union[ExperimentDesign, Experiment, ULID],
-        campaign: Optional[Union[CampaignDesign, Campaign, ULID]] = None,
+        experiment: Union[ExperimentDesign, Experiment, str],
+        campaign: Optional[Union[CampaignDesign, Campaign, str]] = None,
     ):
         """Creates a new experiment, or continues an existing experiment
 
         Parameters
         ----------
-        experiment: Union[ExperimentDesign, Experiment, ULID]
+        experiment: Union[ExperimentDesign, Experiment, str]
             A new experiment design, or an existing experiment or experiment_id to continue
 
-        campaign: Optional[Union[CampaignDesign, Campaign, ULID]]
+        campaign: Optional[Union[CampaignDesign, Campaign, str]]
             A new campaign design, or an existing campaign or campaign_id to associate with the experiment
 
         Returns
@@ -291,7 +292,7 @@ def check_in(self):
 
     def _check_experiment(self):
         """Checks that the experiment has been created"""
-        if self.expeirment_started is None:
+        if self.experiment_started is None:
             if self.experiment is None:
                 raise ValueError(
                     "Experiment has not been set. Please assign a valid experiment before performing this action."
@@ -441,7 +442,7 @@ def validate_workflow(
             validate_only=True,
         )
 
-    def await_runs(self, run_ids: List[str, ULID]) -> Dict[str, WorkflowRun]:
+    def await_runs(self, run_ids: List[Union[str, ULID]]) -> Dict[str, WorkflowRun]:
         """
         Waits for all provided runs to complete, then returns a dictionary of their results
         """
diff --git a/src/wei/routers/campaign_routes.py b/src/wei/routers/campaign_routes.py
index d75ee866..2855f2df 100644
--- a/src/wei/routers/campaign_routes.py
+++ b/src/wei/routers/campaign_routes.py
@@ -34,14 +34,14 @@ def register_campaign(campaign_design: CampaignDesign) -> Campaign:
     return campaign
 
 
-@router.get("/{campaign_id}")
-def get_campaign(campaign_id: str) -> Campaign:
-    """Returns the details of a campaign"""
-    return state_manager.get_campaign(campaign_id)
-
-
 @router.get("/")
 @router.get("/all")
 def get_all_campaigns() -> Dict[str, Campaign]:
     """Returns all campaigns"""
     return state_manager.get_all_campaigns()
+
+
+@router.get("/{campaign_id}")
+def get_campaign(campaign_id: str) -> Campaign:
+    """Returns the details of a campaign"""
+    return state_manager.get_campaign(campaign_id)
diff --git a/src/wei/routers/event_routes.py b/src/wei/routers/event_routes.py
index 3e4f5aab..3c932a54 100644
--- a/src/wei/routers/event_routes.py
+++ b/src/wei/routers/event_routes.py
@@ -2,7 +2,7 @@
 Router for the "events" endpoints
 """
 
-from typing import Any
+from typing import Any, Dict
 
 from fastapi import APIRouter
 
@@ -21,13 +21,14 @@ def log_event(event: Event) -> Any:
     return event
 
 
+@router.get("/")
+@router.get("/all")
+def get_all_events() -> Dict[str, Event]:
+    """Returns all events stored in the event cache."""
+    return state_manager.get_all_events()
+
+
 @router.get("/{event_id}")
 def get_event(event_id: str) -> Event:
     """Returns the details for a specific event given the id"""
     return state_manager.get_event(event_id)
-
-
-@router.get("/")
-def get_all_events(event_id: str) -> Event:
-    """Returns all events stored in the event cache."""
-    return state_manager.get_all_events()
diff --git a/src/wei/types/base_types.py b/src/wei/types/base_types.py
index cdbccf66..3559f3a2 100644
--- a/src/wei/types/base_types.py
+++ b/src/wei/types/base_types.py
@@ -2,12 +2,13 @@
 
 import json
 from pathlib import Path
-from typing import Optional, Type, TypeVar, Union
+from typing import Optional, Self, Type, TypeVar, Union
 
-import ulid
 import yaml
 from pydantic import AliasChoices, Field
 from pydantic import BaseModel as _BaseModel
+from pydantic.functional_validators import model_validator
+from ulid import ULID
 
 _T = TypeVar("_T")
 
@@ -15,8 +16,8 @@
 
 
 def ulid_factory() -> str:
-    """Generates a ulid string"""
-    return ulid.new().str
+    """Generates a ulid"""
+    return str(ULID())
 
 
 class BaseModel(_BaseModel, use_enum_values=True):
@@ -49,6 +50,18 @@ def from_yaml(cls: Type[_T], path: PathLike) -> _T:
             raw_data = yaml.safe_load(fp)
         return cls(**raw_data)
 
+    @model_validator(mode="after")
+    def validate_ulids(self) -> Self:
+        """Validates that all ULID fields are valid"""
+        for field in self.model_fields:
+            if field == "id" or field.endswith("_id"):
+                try:
+                    if self.model_dump()[field]:
+                        ULID.from_str(self.model_dump()[field])
+                except ValueError as e:
+                    raise ValueError(f"Invalid ULID in field {field}") from e
+        return self
+
 
 class Metadata(BaseModel, extra="allow"):
     """Metadata container"""
diff --git a/src/wei/types/datapoint_types.py b/src/wei/types/datapoint_types.py
index 47a39e19..e48308d2 100644
--- a/src/wei/types/datapoint_types.py
+++ b/src/wei/types/datapoint_types.py
@@ -3,7 +3,6 @@
 from typing import Any, Literal, Optional
 
 from pydantic import Field
-from pydantic_extra_types.ulid import ULID
 
 from wei.types.base_types import BaseModel, ulid_factory
 
@@ -23,7 +22,7 @@ class DataPoint(BaseModel, extra="allow"):
     """campaign of the data point"""
     type: str
     """type of the datapoint, inherited from class"""
-    id: ULID = Field(default_factory=ulid_factory)
+    id: str = Field(default_factory=ulid_factory)
     """specific id for this data point"""
 
 
diff --git a/src/wei/types/event_types.py b/src/wei/types/event_types.py
index 6dac5e08..43b422e6 100644
--- a/src/wei/types/event_types.py
+++ b/src/wei/types/event_types.py
@@ -4,7 +4,6 @@
 from typing import Any, Dict, List, Literal, Optional
 
 from pydantic import Field
-from pydantic_extra_types.ulid import ULID
 
 from wei.types.base_types import BaseModel, ulid_factory
 from wei.types.experiment_types import Campaign, Experiment
@@ -16,7 +15,7 @@
 class Event(BaseModel, extra="allow"):
     """A single event in an experiment"""
 
-    event_id: ULID = Field(default_factory=ulid_factory)
+    event_id: str = Field(default_factory=ulid_factory)
     event_timestamp: datetime = Field(default_factory=datetime.now)
     experiment_id: Optional[str] = None
     campaign_id: Optional[str] = None
@@ -139,7 +138,7 @@ class GladierFlowEvent(Event):
 
     flow_name: str
     """The name of the flow"""
-    flow_id: ULID
+    flow_id: str
     """The ID of the flow"""
 
 
@@ -192,7 +191,7 @@ class WorkflowQueuedEvent(Event):
     event_type: Literal["WORKFLOW"] = "WORKFLOW"
     event_name: Literal["QUEUED"] = "QUEUED"
 
-    run_id: ULID
+    run_id: str
     """The ID of the workflow"""
     workflow_name: str
     """The name of the workflow"""
@@ -213,7 +212,7 @@ class WorkflowStartEvent(Event):
     event_type: Literal["WORKFLOW"] = "WORKFLOW"
     event_name: Literal["START"] = "START"
 
-    run_id: ULID
+    run_id: str
     """The ID of the workflow"""
     workflow_name: str
     """The name of the workflow"""
@@ -234,7 +233,7 @@ class WorkflowFailedEvent(Event):
     event_type: Literal["WORKFLOW"] = "WORKFLOW"
     event_name: Literal["FAILED"] = "FAILED"
 
-    run_id: ULID
+    run_id: str
     """The ID of the workflow"""
     workflow_name: str
     """The name of the workflow"""
@@ -255,7 +254,7 @@ class WorkflowCompletedEvent(Event):
     event_type: Literal["WORKFLOW"] = "WORKFLOW"
     event_name: Literal["COMPLETED"] = "COMPLETED"
 
-    run_id: ULID
+    run_id: str
     """The ID of the workflow"""
     workflow_name: str
     """The name of the workflow"""
@@ -276,7 +275,7 @@ class WorkflowCancelled(Event):
     event_type: Literal["WORKFLOW"] = "WORKFLOW"
     event_name: Literal["CANCELLED"] = "CANCELLED"
 
-    run_id: ULID
+    run_id: str
     """The ID of the workflow"""
     workflow_name: str
     """The name of the workflow"""
@@ -297,7 +296,7 @@ class WorkflowPausedEvent(Event):
     event_type: Literal["WORKFLOW"] = "WORKFLOW"
     event_name: Literal["PAUSED"] = "PAUSED"
 
-    run_id: ULID
+    run_id: str
     """The ID of the workflow"""
     workflow_name: str
     """The name of the workflow"""
@@ -318,7 +317,7 @@ class WorkflowResumedEvent(Event):
     event_type: Literal["WORKFLOW"] = "WORKFLOW"
     event_name: Literal["RESUMED"] = "RESUMED"
 
-    run_id: ULID
+    run_id: str
     """The ID of the workflow"""
     workflow_name: str
     """The name of the workflow"""
@@ -339,7 +338,7 @@ class WorkflowStepEvent(Event):
     event_type: Literal["WORKFLOW"] = "WORKFLOW"
     event_name: Literal["STEP"] = "STEP"
 
-    run_id: ULID
+    run_id: str
     """The ID of the workflow"""
     workflow_name: str
     """The name of the workflow"""
diff --git a/src/wei/types/experiment_types.py b/src/wei/types/experiment_types.py
index 9fd54423..55f9b018 100644
--- a/src/wei/types/experiment_types.py
+++ b/src/wei/types/experiment_types.py
@@ -4,7 +4,6 @@
 from typing import List, Optional
 
 from pydantic import AliasChoices, Field
-from pydantic_extra_types.ulid import ULID
 
 from wei.types.base_types import BaseModel, PathLike, ulid_factory
 
@@ -14,14 +13,14 @@ class CampaignDesign(BaseModel):
 
     campaign_name: str
     """Name of the campaign"""
-    description: Optional[str] = None
+    campaign_description: Optional[str] = None
     """Description of the campaign"""
 
 
 class Campaign(CampaignDesign):
     """A campaign is a collection of related experiments"""
 
-    campaign_id: ULID = Field(default_factory=ulid_factory)
+    campaign_id: str = Field(default_factory=ulid_factory)
     """ID of the campaign"""
     experiment_ids: List[str] = []
     """Experiments associated with the campaign"""
@@ -34,16 +33,16 @@ class ExperimentDesign(BaseModel):
     """Name of the experiment"""
     campaign_id: Optional[str] = None
     """ID of the campaign this experiment should be associated with (note: this campaign must already exist)"""
-    description: Optional[str] = None
+    experiment_description: Optional[str] = None
     """Description of the experiment"""
     email_addresses: List[str] = []
-    """List of email addresses to send notifications"""
+    """List of email addresses to send notifications to"""
 
 
 class Experiment(ExperimentDesign):
     """A single instance of an experiment"""
 
-    experiment_id: ULID = Field(default_factory=ulid_factory)
+    experiment_id: str = Field(default_factory=ulid_factory)
     """ID of the experiment"""
     experiment_directory: Optional[PathLike] = None
     """The directory where the experiment is stored on disk"""
diff --git a/src/wei/types/module_types.py b/src/wei/types/module_types.py
index 92aa8328..eae0de2f 100644
--- a/src/wei/types/module_types.py
+++ b/src/wei/types/module_types.py
@@ -11,7 +11,6 @@
     field_validator,
     model_validator,
 )
-from pydantic_extra_types.ulid import ULID
 from typing_extensions import Literal, Self
 
 from wei.types.base_types import BaseModel, ulid_factory
@@ -253,7 +252,7 @@ def validate_config_against_interface(self) -> Self:
 class Module(ModuleDefinition):
     """Live instance of a Module"""
 
-    id: ULID = Field(default_factory=ulid_factory)
+    id: str = Field(default_factory=ulid_factory)
     """ID of this instance of a Module"""
     state: ModuleState = Field(default=ModuleState(status=ModuleStatus.UNKNOWN))
     """Current state of the module"""
diff --git a/src/wei/types/step_types.py b/src/wei/types/step_types.py
index 377c8985..65a8f8bb 100644
--- a/src/wei/types/step_types.py
+++ b/src/wei/types/step_types.py
@@ -10,8 +10,7 @@
 import yaml
 from fastapi import UploadFile
 from fastapi.responses import FileResponse
-from pydantic import AliasChoices, Field, ValidationInfo, field_validator, validator
-from pydantic_extra_types.ulid import ULID
+from pydantic import AliasChoices, Field, ValidationInfo, field_validator
 from typing_extensions import Literal
 
 from wei.types.base_types import BaseModel, PathLike, ulid_factory
@@ -155,7 +154,7 @@ class Step(BaseModel, arbitrary_types_allowed=True):
 
     """Runtime information"""
 
-    id: ULID = Field(default_factory=ulid_factory)
+    id: str = Field(default_factory=ulid_factory)
     """ID of step"""
     start_time: Optional[datetime] = None
     """Time the step started running"""
@@ -167,8 +166,9 @@ class Step(BaseModel, arbitrary_types_allowed=True):
     """Result of the step after being run"""
 
     # Load any yaml arguments
-    @validator("args")
-    def validate_args_dict(cls, v: Any, **kwargs: Any) -> Any:
+    @field_validator("args")
+    @classmethod
+    def validate_args_dict(cls, v: Any) -> Any:
         """asserts that args dict is assembled correctly"""
         assert isinstance(v, dict), "Args is not a dictionary"
         for key, arg_data in v.items():
diff --git a/src/wei/types/workflow_types.py b/src/wei/types/workflow_types.py
index 3a2dabf9..630d38f6 100644
--- a/src/wei/types/workflow_types.py
+++ b/src/wei/types/workflow_types.py
@@ -5,7 +5,6 @@
 from typing import Any, Dict, List, Optional, Union
 
 from pydantic import Field, field_validator
-from pydantic_extra_types.ulid import ULID
 
 from wei.types.base_types import BaseModel, Metadata, ulid_factory
 from wei.types.module_types import SimpleModule
@@ -88,7 +87,7 @@ class WorkflowRun(Workflow):
 
     label: Optional[str] = None
     """Label for the workflow run"""
-    run_id: ULID = Field(default_factory=ulid_factory)
+    run_id: str = Field(default_factory=ulid_factory)
     """ID of the workflow run"""
     payload: Dict[str, Any] = {}
     """input information for a given workflow run"""
@@ -98,7 +97,7 @@ class WorkflowRun(Workflow):
     """WEI Processed Steps of the flow"""
     hist: Dict[str, Any] = Field(default={})
     """history of the workflow"""
-    experiment_id: ULID
+    experiment_id: str
     """ID of the experiment this workflow is a part of"""
     step_index: int = 0
     """Index of the current step"""
@@ -118,7 +117,7 @@ def get_step_by_name(self, name: str) -> Step:
                 return step
         raise KeyError(f"Step {name} not found in workflow run {self.run_id}")
 
-    def get_step_by_id(self, id: ULID) -> Step:
+    def get_step_by_id(self, id: str) -> Step:
         """Return the step object indexed by its id"""
         for step in self.steps:
             if step.id == id:
diff --git a/tests/test_base.py b/tests/test_base.py
index ea8da162..3958d52f 100644
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -14,12 +14,12 @@ class TestWEI_Base(unittest.TestCase):
 
     experiment = ExperimentDesign(
         experiment_name="Test_Experiment",
-        description="An experiment for automated testing",
+        experiment_description="An experiment for automated testing",
         email_addresses=[],
     )
     campaign = CampaignDesign(
         campaign_name="Test_Campaign",
-        description="A campaign for automated testing",
+        campaign_description="A campaign for automated testing",
     )
 
     def __init__(self, *args, **kwargs):