diff --git a/backend/build_binary.py b/backend/build_binary.py
index 5f43652a..c4868bed 100644
--- a/backend/build_binary.py
+++ b/backend/build_binary.py
@@ -27,14 +27,14 @@ def build_server():
         print(f"Using local qwen_tts source from: {local_qwen_path}")
 
     # Exclude unnecessary modules to reduce size
+    # Be careful with torch exclusions - torch internally imports many of its own submodules
     args.extend([
         '--exclude-module', 'matplotlib',
         '--exclude-module', 'IPython',
         '--exclude-module', 'notebook',
         '--exclude-module', 'pytest',
         # Don't exclude setuptools - pkg_resources needs it
-        '--exclude-module', 'torch.distributions',
-        '--exclude-module', 'torch.testing',
+        # Don't exclude torch.testing - torch.autograd.gradcheck needs it
         '--exclude-module', 'tensorboard',
         '--exclude-module', 'torch.utils.tensorboard',
         '--exclude-module', 'scipy',