diff --git a/src/main.py b/src/main.py
index 7edccf7..2dd9758 100644
--- a/src/main.py
+++ b/src/main.py
@@ -161,7 +161,7 @@ if __name__ == '__main__':
     requiredNamed.add_argument('-d', '--dataset', help='Name of the dataset', required=True, type=str)
     opt = parser.parse_args()
 
-    assert opt.dataset in ['enron', 'imdb62', 'blogs'], 'unknown dataset'
+    assert opt.dataset in ['enron', 'imdb62', 'blogs', 'victorian'], 'unknown dataset'
 
     create_path_if_not_exists(opt.output)
     create_path_if_not_exists(opt.log)