diff --git a/eolearn/features/__init__.py b/eolearn/features/__init__.py
index 8b446c0e..34c9eece 100644
--- a/eolearn/features/__init__.py
+++ b/eolearn/features/__init__.py
@@ -2,5 +2,5 @@
 A collection of EOTasks for feature manipulation
 """
 
-from .bands_extraction import NormalizedDifferenceIndexTask
 from .feature_manipulation import FilterTimeSeriesTask, SimpleFilterTask
+from .ndi import NormalizedDifferenceIndexTask
diff --git a/eolearn/features/bands_extraction.py b/eolearn/features/ndi.py
similarity index 100%
rename from eolearn/features/bands_extraction.py
rename to eolearn/features/ndi.py