diff --git a/python/upload_templates.py b/python/upload_templates.py
index 66e2adf..5f4f5d7 100644
--- a/python/upload_templates.py
+++ b/python/upload_templates.py
@@ -259,7 +259,7 @@ def find_channels_with_max_peak_to_peak_vectorized(templates):
         # Do a check for the expected shape of the templates
         number_of_units = sorting.get_num_units()
         number_of_temporal_samples = target_nbefore + target_nafter
-        number_of_channels = pre_processed_recording.get_number_of_channels()
+        number_of_channels = pre_processed_recording.get_num_channels()
         expected_shape = (number_of_units, number_of_temporal_samples, number_of_channels)
         assert templates_extension_data.templates_array.shape == expected_shape