ericgj
diff --git a/‎functions/ocr/app/main.py
Copy file name to clipboardExpand all lines: functions/ocr/app/main.py
+3-3Lines changed: 3 additions & 3 deletions b/‎functions/ocr/app/main.py
Copy file name to clipboardExpand all lines: functions/ocr/app/main.py
+3-3Lines changed: 3 additions & 3 deletions
diff --git a/‎functions/ocr/app/main_test.py
Copy file name to clipboardExpand all lines: functions/ocr/app/main_test.py
+1-1Lines changed: 1 addition & 1 deletion b/‎functions/ocr/app/main_test.py
Copy file name to clipboardExpand all lines: functions/ocr/app/main_test.py
+1-1Lines changed: 1 addition & 1 deletion
@@ -57,12 +57,12 @@ def detect_text(bucket, filename):
     # Submit a message to the bus for each target language
     for target_lang in config.get('TO_LANG', []):
         topic_name = config['TRANSLATE_TOPIC']
-        if src_lang == target_lang:
+        if src_lang == target_lang or src_lang == 'und':
             topic_name = config['RESULT_TOPIC']
         message = {
             'text': text,
             'filename': filename,
-            'target_lang': target_lang,
+            'lang': target_lang,
             'src_lang': src_lang
         }
         message_data = json.dumps(message).encode('utf-8')
@@ -113,7 +113,7 @@ def translate_text(event, context):
 
     text = validate_message(message, 'text')
     filename = validate_message(message, 'filename')
-    target_lang = validate_message(message, 'target_lang')
+    target_lang = validate_message(message, 'lang')
     src_lang = validate_message(message, 'src_lang')
 
     print('Translating text into {}.'.format(target_lang))
 
@@ -62,7 +62,7 @@ def test_translate_text(self, mock_translate_client, mock_publisher):
         data = base64.b64encode(json.dumps({
             'text': 'menu',
             'filename': 'sample-file',
-            'target_lang': 'es',
+            'lang': 'es',
             'src_lang': 'en'
         }).encode('utf-8'))
         event = {