fix: Fix module configuration loading (#487)

1 files changed, 12 insertions, 7 deletions

diff --git a/src/module.rs b/src/module.rs
index d14814eb0..b06bf5529 100644
--- a/src/module.rs
+++ b/src/module.rs
@@ -80,15 +80,20 @@ impl<'a> Module<'a> {
         let mut segment = Segment::new(name);
         let segment_config_mock = SegmentConfig { value, style: None };
 
-        if let Some(module_config) = self.config {
-            let segment_config = segment_config_mock.load_config(&module_config);
-            segment.set_style(segment_config.style.unwrap_or(self.style));
-            segment.set_value(segment_config.value);
-        } else {
-            segment.set_style(segment_config_mock.style.unwrap_or(self.style));
-            segment.set_value(segment_config_mock.value);
+        if let Some(toml::Value::Table(module_config)) = self.config {
+            if let Some(symbol) = module_config.get(name) {
+                let segment_config = segment_config_mock.load_config(&symbol);
+                segment.set_style(segment_config.style.unwrap_or(self.style));
+                segment.set_value(segment_config.value);
+
+                self.segments.push(segment);
+                return self.segments.last_mut().unwrap();
+            }
         }
 
+        segment.set_style(segment_config_mock.style.unwrap_or(self.style));
+        segment.set_value(segment_config_mock.value);
+
         self.segments.push(segment);
         self.segments.last_mut().unwrap()
     }