refactor folder strucure

3 files changed, 4 insertions, 3 deletions

diff --git a/examples/cc_cpe_labeling.py b/examples/cc_cpe_labeling.py
index fa96a575..c0cfe6ab 100644
--- a/examples/cc_cpe_labeling.py
+++ b/examples/cc_cpe_labeling.py
@@ -2,7 +2,7 @@ from datetime import datetime
 import logging
 from pathlib import Path
 
-from sec_certs.dataset import CCDataset
+from sec_certs.dataset.common_criteria import CCDataset
 import sec_certs.constants as constants
 
 logger = logging.getLogger(__name__)
diff --git a/examples/cc_oop_demo.py b/examples/cc_oop_demo.py
index a78fe147..77acebb1 100644
--- a/examples/cc_oop_demo.py
+++ b/examples/cc_oop_demo.py
@@ -1,4 +1,4 @@
-from sec_certs.dataset import CCDataset
+from sec_certs.dataset.common_criteria import CCDataset
 from sec_certs.serialization import CustomJSONEncoder, CustomJSONDecoder
 import sec_certs.constants as constants
 from pathlib import Path
diff --git a/examples/fips_oop_demo.py b/examples/fips_oop_demo.py
index 7015574e..dde76101 100644
--- a/examples/fips_oop_demo.py
+++ b/examples/fips_oop_demo.py
@@ -2,7 +2,8 @@ from pathlib import Path
 from datetime import datetime
 import logging
 import click
-from sec_certs.dataset import FIPSDataset, FIPSAlgorithmDataset
+from sec_certs.dataset.fips import FIPSDataset
+from sec_certs.dataset.fips_algorithm import FIPSAlgorithmDataset
 from sec_certs.configuration import config
 from sec_certs.helpers import analyze_matched_algs