diff --git a/python/otbtf.py b/python/otbtf.py
index a23d5237b701b9ca245fa9b850162ca2ec52e064..860e86a7b30dc182698ec521a341daeb44d00c99 100644
--- a/python/otbtf.py
+++ b/python/otbtf.py
@@ -2,7 +2,7 @@
 # ==========================================================================
 #
 #   Copyright 2018-2019 IRSTEA
-#   Copyright 2020-2021 INRAE
+#   Copyright 2020-2022 INRAE
 #
 #   Licensed under the Apache License, Version 2.0 (the "License");
 #   you may not use this file except in compliance with the License.
@@ -28,7 +28,7 @@ import logging
 from abc import ABC, abstractmethod
 import numpy as np
 import tensorflow as tf
-import gdal
+from osgeo import gdal
 
 
 # ----------------------------------------------------- Helpers --------------------------------------------------------