diff --git a/formatters.py b/formatters.py
index 9d30c55..fc3bd4c 100644
--- a/formatters.py
+++ b/formatters.py
@@ -4,7 +4,7 @@
 
 
 def get_formatter(test_type):
-    if test_type == "io":
+    if test_type == "iozone" or test_type == "fio":
         return format_io_stat
     elif test_type == "pgbench":
         return format_pgbench_stat
@@ -66,8 +66,3 @@
             data[key] = (mean, dev)
         return data
 
-
-
-
-
-