diff --git a/wally/run_test.py b/wally/run_test.py
index 2cd397e..af317b3 100755
--- a/wally/run_test.py
+++ b/wally/run_test.py
@@ -26,12 +26,14 @@
 from wally.suits.itest import TestConfig
 from wally.suits.io.fio import IOPerfTest
 from wally.suits.postgres import PgBenchTest
+from wally.suits.omgbench import OmgTest
 
 
 TOOL_TYPE_MAPPER = {
     "io": IOPerfTest,
     "pgbench": PgBenchTest,
     "mysql": MysqlTest,
+    "omg": OmgTest,
 }
 
 
@@ -540,6 +542,8 @@
                 rep = "\n\n".join(rep_lst)
             elif tp in ['mysql', 'pgbench'] and data is not None:
                 rep = MysqlTest.format_for_console(data)
+            elif tp == 'omg':
+                rep = OmgTest.format_for_console(data)
             else:
                 logger.warning("Can't generate text report for " + tp)
                 continue