diff --git a/matsciml/models/losses.py b/matsciml/models/losses.py
index 207da6d2..75ce57e0 100644
--- a/matsciml/models/losses.py
+++ b/matsciml/models/losses.py
@@ -6,7 +6,7 @@
 from torch import nn
 
 
-__all__ = ["AtomWeightedL1", "AtomWeightedMSE"]
+__all__ = ["AtomWeightedL1", "AtomWeightedMSE", "BatchQuantileLoss"]
 
 
 class AtomWeightedL1(nn.Module):