diff --git a/aimodel/src/lib/ai/components/LossDice.py b/aimodel/src/lib/ai/components/LossDice.py
index 2f5c0c5..0df4baa 100644
--- a/aimodel/src/lib/ai/components/LossDice.py
+++ b/aimodel/src/lib/ai/components/LossDice.py
@@ -35,6 +35,7 @@ def dice_coef_loss(y_true, y_pred, **kwargs):
 
 class LossDice(tf.keras.losses.Loss):
 	"""An implementation of the dice loss function.
+	@source 
 	Args:
 		smooth (float): The batch size (currently unused).
 	"""