diff --git a/tito/commands/info.py b/tito/commands/info.py
index 21f25ee7..2ee4279a 100644
--- a/tito/commands/info.py
+++ b/tito/commands/info.py
@@ -78,7 +78,7 @@ class InfoCommand(BaseCommand):
             ("Networks", "neural network architectures", self.check_networks_status),
             ("MLP", "multi-layer perceptron (manual)", self.check_mlp_status),
             ("CNN", "convolutional networks (basic)", self.check_cnn_status),
-            ("Data", "data loading pipeline", self.check_data_status),
+            ("DataLoader", "data loading pipeline", self.check_data_status),
             ("Training", "autograd engine & optimization", self.check_training_status),
             ("Profiling", "performance profiling", self.check_profiling_status),
             ("Compression", "model compression", self.check_compression_status),
diff --git a/tito/commands/test.py b/tito/commands/test.py
index 14d05612..51da77ec 100644
--- a/tito/commands/test.py
+++ b/tito/commands/test.py
@@ -32,7 +32,7 @@ class TestCommand(BaseCommand):
 
     def run(self, args: Namespace) -> int:
         console = self.console
-        valid_modules = ["setup", "tensor", "activations", "layers", "networks", "cnn", "data", "training", 
+        valid_modules = ["setup", "tensor", "activations", "layers", "networks", "cnn", "dataloader", "training", 
                          "profiling", "compression", "kernels", "benchmarking", "mlops"]
         
         if args.all: