diff --git a/indica/bayesmodels.py b/indica/bayesmodels.py
index 900946a6..a05e8c77 100644
--- a/indica/bayesmodels.py
+++ b/indica/bayesmodels.py
@@ -194,7 +194,7 @@ def sample_from_priors(self, param_names, size=10):
         return samples.transpose()
 
     def sample_from_high_density_region(
-        self, param_names: list, sampler: object, nwalkers: int, nsamples=100
+        self, param_names: list, sampler, nwalkers: int, nsamples=100
     ):
         start_points = self.sample_from_priors(param_names, size=nsamples)
 
diff --git a/indica/workflows/abstract_bayes_workflow.py b/indica/workflows/abstract_bayes_workflow.py
index 4b7d2ef8..3e83fd4e 100644
--- a/indica/workflows/abstract_bayes_workflow.py
+++ b/indica/workflows/abstract_bayes_workflow.py
@@ -34,14 +34,13 @@ def __init__(
     def read_test_data(self, diagnostic_transforms: dict, tstart=None, tend=None, dt=None):
         # Used with phantom data for purposes of tests
         print("Reading fake data")
-        self.reader = None
         self.equilibrium = fake_equilibrium(
             tstart,
             tend,
             dt,
         )
         self.transforms = diagnostic_transforms
-        self.data = {}
+        self.data: dict = {}
 
     def read_data(self, diagnostics: list, tstart=None, tend=None, dt=None):
         self.reader = ReadST40(
@@ -67,7 +66,7 @@ def setup_models(self, diagnostics: list):
         Initialising models normally requires data to be read so transforms can be set
 
         """
-        self.models = {}
+        self.models: dict = {}
 
     @abstractmethod
     def _phantom_data(self):