diff --git a/speedtest_sync.py b/speedtest_sync.py
index 3477d3b..b7c4b19 100644
--- a/speedtest_sync.py
+++ b/speedtest_sync.py
@@ -135,8 +135,8 @@ invoice_files = [
     ('invoice_file', ('invoice.jpg', open("test_samples/sbt/invoice.jpg", "rb").read())),
 ]
 imei_files = [
-    ('imei_files', ("test_samples/sbt/imei1.jpg", open("test_samples/sbt/imei1.jpg", "rb").read())),
-    # ('imei_files', ("test_samples/sbt/imei2.jpg", open("test_samples/sbt/imei2.jpg", "rb").read())),
+    ('imei_files', ("test_samples/sbt/imei1.jpg", open("test_samples/sbt/imei1.jpg", "rb").read(), 'application/octet-stream')),
+    ('imei_files', ("test_samples/sbt/imei2.jpg", open("test_samples/sbt/imei1.jpg", "rb").read(), 'application/octet-stream')),
     # ('imei_files', ("test_samples/sbt/imei3.jpg", open("test_samples/sbt/imei3.jpg", "rb").read())),
     # ('imei_files', ("test_samples/sbt/imei4.jpeg", open("test_samples/sbt/imei4.jpeg", "rb").read())),
     # ('imei_files', ("test_samples/sbt/imei5.jpg", open("test_samples/sbt/imei5.jpg", "rb").read())),
@@ -146,8 +146,15 @@ def get_imei_files():
     files = imei_files[:num_files]
     # print("Num of imei files:", len(files))
     return files
+# def get_files():
+#     return imei_files
+    # return invoice_files + get_imei_files()
 def get_files():
-    return invoice_files + imei_files # get_imei_files()
+    return {
+        'invoice_file': ('invoice.jpg', open("test_samples/sbt/invoice.jpg", "rb").read()),
+        'imei_files': ("test_samples/sbt/imei1.jpg", open("test_samples/sbt/imei1.jpg", "rb").read(), 'application/octet-stream'),
+        'processType': 12,
+    }
 def gen_input(num_input):
     for _ in range(num_input):
         yield (None, token)