diff --git a/android/test_app/make_assets.py b/android/test_app/make_assets.py
index ce5059b46ee..7860c759a57 100644
--- a/android/test_app/make_assets.py
+++ b/android/test_app/make_assets.py
@@ -1,5 +1,6 @@
 import torch
 import torchvision
+from torch.utils.mobile_optimizer import optimize_for_mobile
 
 print(torch.__version__)
 
@@ -12,6 +13,5 @@
 
 model.eval()
 script_model = torch.jit.script(model)
-# TODO: put back call to optimize_for_mobile once
-# https://github.com/pytorch/pytorch/issues/52463 is fixed
-script_model.save("app/src/main/assets/frcnn_mnetv3.pt")
+opt_script_model = optimize_for_mobile(script_model)
+opt_script_model.save("app/src/main/assets/frcnn_mnetv3.pt")