diff --git a/LeanBandits/ForMathlib/Integrable.lean b/LeanBandits/ForMathlib/Integrable.lean
index 8feeb38f..5a366cf3 100644
--- a/LeanBandits/ForMathlib/Integrable.lean
+++ b/LeanBandits/ForMathlib/Integrable.lean
@@ -3,7 +3,7 @@ Copyright (c) 2026 Rémy Degenne. All rights reserved.
 Released under Apache 2.0 license as described in the file LICENSE.
 Authors: Rémy Degenne
 -/
-import Mathlib
+import Mathlib.Probability.IdentDistrib
 
 open ProbabilityTheory