diff --git a/ref-test/app/models/dataset.py b/ref-test/app/models/dataset.py
index d3dc485..9fcc249 100644
--- a/ref-test/app/models/dataset.py
+++ b/ref-test/app/models/dataset.py
@@ -102,7 +102,7 @@ class Dataset(db.Model):
         with open(file_path, 'w') as file:
             dump(data, file, indent=2)
         write('system.log', f'Dataset {self.id} edited by {current_user.get_username()}.')
-        flash(f'Dataset {self.name} successfully edited.', 'success')
+        flash(f'Dataset {self.get_name()} successfully edited.', 'success')
         db.session.add(self)
         db.session.commit()
         return True, 'Dataset successfully edited.'
\ No newline at end of file