diff --git a/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java b/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java
index c971aa96..e7654212 100644
--- a/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java
+++ b/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java
@@ -2969,6 +2969,13 @@ public void testModule77()
         new int[] {2});
   }
 
+  /** Test https://github.com/wala/ML/issues/211. */
+  @Test
+  public void testModule78()
+      throws ClassHierarchyException, IllegalArgumentException, CancelException, IOException {
+    test(new String[] {"module.py", "client2.py"}, "module.py", "f", "", 1, 1, new int[] {2});
+  }
+
   @Test
   public void testStaticMethod() throws ClassHierarchyException, CancelException, IOException {
     test("tf2_test_static_method.py", "MyClass.the_static_method", 1, 1, 2);
diff --git a/com.ibm.wala.cast.python.test/data/client2.py b/com.ibm.wala.cast.python.test/data/client2.py
new file mode 100644
index 00000000..02346a3f
--- /dev/null
+++ b/com.ibm.wala.cast.python.test/data/client2.py
@@ -0,0 +1,6 @@
+# Test https://github.com/wala/ML/issues/211.
+
+from tensorflow import ones
+from module import f
+
+f(ones([1, 2]))