diff --git a/src/pytorch/OptimizedTorchANI.py b/src/pytorch/OptimizedTorchANI.py
index ef8a451..309eab0 100644
--- a/src/pytorch/OptimizedTorchANI.py
+++ b/src/pytorch/OptimizedTorchANI.py
@@ -32,7 +32,7 @@
 
 class OptimizedTorchANI(torch.nn.Module):
 
-    from torchani.models import BuiltinModel
+    from torchani.models import BuiltinModel # https://github.com/openmm/NNPOps/issues/44
 
     def __init__(self, model: BuiltinModel, atomicNumbers: Tensor) -> None: