diff --git a/forge/test/mlir/test_ops.py b/forge/test/mlir/test_ops.py
index 9ee8b3fe..80692811 100644
--- a/forge/test/mlir/test_ops.py
+++ b/forge/test/mlir/test_ops.py
@@ -354,7 +354,7 @@ def forward(self, x):
 @pytest.mark.parametrize("token_num", [12])
 @pytest.mark.parametrize("embedding_dim", [3200])
 def test_embedding(vocab_size, token_num, embedding_dim):
-    compiler_cfg = pyforge.config._get_global_compiler_config()
+    compiler_cfg = forge.config._get_global_compiler_config()
     compiler_cfg.enable_tvm_cpu_fallback = False
 
     class Embedding(nn.Module):
@@ -372,14 +372,13 @@ def forward(self, x):
     framework_model = Embedding()
     fw_out = framework_model(*inputs)
 
-    compiled_model = pyforge.compile(framework_model, sample_inputs=inputs)
+    compiled_model = forge.compile(framework_model, sample_inputs=inputs)
     co_out = compiled_model(*inputs)
 
     co_out = [co.to("cpu") for co in co_out]
     assert compare_with_golden_pcc(golden=fw_out, calculated=co_out[0], pcc=0.99)
 
 
-@pytest.mark.xfail(reason="Program expects 2 inputs, found 1 in input tensors vector")
 @pytest.mark.parametrize("shape", [
     (7,),           # 1D tensor
     (32,),          # 1D tensor
@@ -405,7 +404,7 @@ def forward(self, x):
     framework_model = Reciprocal()
     fw_out = framework_model(*inputs)
     
-    compiled_model = pyforge.compile(framework_model, sample_inputs=inputs)
+    compiled_model = forge.compile(framework_model, sample_inputs=inputs)
     co_out = compiled_model(*inputs)
     
     co_out = [co.to("cpu") for co in co_out]
diff --git a/third_party/tvm b/third_party/tvm
index 736a40ab..416060a1 160000
--- a/third_party/tvm
+++ b/third_party/tvm
@@ -1 +1 @@
-Subproject commit 736a40ab35ed3b1aa8d24955cf0b497412d2866d
+Subproject commit 416060a13300ab2b60a4d5ec7c49edbb7d7787a9