diff --git a/pyproject.toml b/pyproject.toml
index ffdd863..e46398d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -59,7 +59,7 @@ docs = [
 "flexpart.cosmo" = "aiida_flexpart.calculations.flexpart_cosmo:FlexpartCosmoCalculation"
 "flexpart.ifs" = "aiida_flexpart.calculations.flexpart_ifs:FlexpartIfsCalculation"
 "flexpart.post" = "aiida_flexpart.calculations.flexpart_post:PostProcessingCalculation"
-"collect.sensitivities" = "aiida_flexpart.calculations.collect_sens:CollectSensCalculation"
+"collect.sensitivities" = "aiida_flexpart.calculations.collect_sens:CollectSensitivitiesCalculation"
 "inversion.calc" = "aiida_flexpart.calculations.inversion:Inversion"
 
 [project.entry-points."aiida.parsers"]