diff --git a/aosstower/level_00/influxdb.py b/aosstower/level_00/influxdb.py
index 4abc566d78b7262a76a598f7270e338f89a5d0b6..8967b98356ade393ac5f453a8f3299d8bb50d366 100644
--- a/aosstower/level_00/influxdb.py
+++ b/aosstower/level_00/influxdb.py
@@ -110,8 +110,8 @@ def main():
     #     wu_pw = open(args.wu_password_file, 'r').read().strip()
 
     if args.ldmp:
-        from aosstower.level_00.parser import ldmp_generator
-        record_gen = ldmp_generator(args.src, args.tables)
+        from aosstower.level_00.parser import LDMPGenerator
+        record_gen = LDMPGenerator(args.src, args.tables)
     else:
         src = open(args.src, "r")
         record_gen = read_frames(src, tail=args.tail)