diff --git a/speos/preprocessing/handler.py b/speos/preprocessing/handler.py
index 8afa81f..86b44a0 100644
--- a/speos/preprocessing/handler.py
+++ b/speos/preprocessing/handler.py
@@ -11,7 +11,7 @@ def __init__(self, config, preprocessor_kwargs):
 
         """
         self.config = config
-        mappings = GWASMapper().get_mappings(config.input.tag, fields=config.input.field)
+        mappings = GWASMapper(self.config.input.gwas_mappings).get_mappings(config.input.tag, fields=config.input.field)
 
         adjacencies = AdjacencyMapper(config.input.adjacency_mappings, blacklist=self.config.input.adjacency_blacklist).get_mappings(config.input.adjacency, fields=config.input.adjacency_field)
 
diff --git a/speos/tests/test_models.py b/speos/tests/test_models.py
index af0b042..a070955 100644
--- a/speos/tests/test_models.py
+++ b/speos/tests/test_models.py
@@ -20,6 +20,7 @@ def setUp(self):
         super().setUp()
 
         self.config.name = "SimpleModelTest"
+        self.config.input.tag = "dummy"
         self.config.training.pos_weight = 2
         self.config.training.dilution = 2
         #self.config.model.mp.type = "gcn"