diff --git a/demo/dynamic_embedding/movielens-1m-keras-with-horovod/movielens-1m-keras-with-horovod.py b/demo/dynamic_embedding/movielens-1m-keras-with-horovod/movielens-1m-keras-with-horovod.py
index a52b585c..5eeaa307 100644
--- a/demo/dynamic_embedding/movielens-1m-keras-with-horovod/movielens-1m-keras-with-horovod.py
+++ b/demo/dynamic_embedding/movielens-1m-keras-with-horovod/movielens-1m-keras-with-horovod.py
@@ -419,7 +419,7 @@ def call(self, features):
             boundaries=fea_info['boundaries'])(input_tensor)
       # To prepare for GPU table combined queries, use a prefix to distinguish different features in a table.
       if fea_info['ptype'] == 'user_occupation_label':
-        input_tensor_prefix_code = int(fea_info['code']) << 1
+        input_tensor_prefix_code = int(fea_info['code']) << 48
       else:
         input_tensor_prefix_code = None
       if input_tensor_prefix_code is not None: