diff --git a/openvalidators/utils.py b/openvalidators/utils.py
index d008f55..e9e6c92 100644
--- a/openvalidators/utils.py
+++ b/openvalidators/utils.py
@@ -54,7 +54,7 @@ def init_wandb(self, reinit=False):
         reinit=reinit,
         project=self.config.wandb.project_name,
         entity=self.config.wandb.entity,
-        config=self.config,
+        config={key: self.config.get(key, None) for key in ('neuron', 'reward')},
         mode="offline" if self.config.wandb.offline else "online",
         dir=self.config.neuron.full_path,
         tags=tags,