diff --git a/MANIFEST.in b/MANIFEST.in
new file mode 100644
index 000000000..2553e20ec
--- /dev/null
+++ b/MANIFEST.in
@@ -0,0 +1 @@
+recursive-include pm4py *.png
\ No newline at end of file
diff --git a/pm4py/llm.py b/pm4py/llm.py
index a49547cb5..faa268556 100644
--- a/pm4py/llm.py
+++ b/pm4py/llm.py
@@ -17,8 +17,8 @@
 
 import pandas as pd
 from pm4py.objects.log.obj import EventLog, EventStream, Trace
-from typing import Union, Optional, Dict, Tuple, Any
-from pm4py.utils import get_properties, constants
+from typing import Union, Optional, Dict, Tuple, List, Any
+from pm4py.utils import get_properties, constants, check_is_pandas_dataframe
 from pm4py.utils import __event_log_deprecation_warning
 from pm4py.objects.ocel.obj import OCEL
 from sqlite3 import Connection as SQ3_Connection