diff --git a/.github/workflows/pythainlp-test.yml b/.github/workflows/pythainlp-test.yml
index d397665f9..8b185a2b0 100644
--- a/.github/workflows/pythainlp-test.yml
+++ b/.github/workflows/pythainlp-test.yml
@@ -23,6 +23,7 @@ jobs:
       run: |
         python -m pip install --upgrade pip pytest wheel flake8
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
+        pip install tensorflow==2.1.0
         pip install torch==1.5.0 torchvision==0.6.0 -f https://download.pytorch.org/whl/torch_stable.html
         pip install .[full]
         pip install deepcut coverage coveralls
diff --git a/setup.py b/setup.py
index 3befb9020..e2b0893eb 100644
--- a/setup.py
+++ b/setup.py
@@ -82,28 +82,11 @@
     test_suite="tests",
     python_requires=">=3.6",
     package_data={
-        "pythainlp.corpus": [
-            "corpus_license.md",
-            "countries_th.txt",
-            "etcc.txt",
-            "negations_th.txt",
-            "person_names_female_th.txt",
-            "person_names_male_th.txt",
-            "pos_orchid_perceptron.pkl",
-            "pos_orchid_unigram.json",
-            "pos_ud_perceptron.pkl",
-            "pos_ud_unigram.json",
-            "sentenceseg_crfcut.model",
-            "stopwords_th.txt",
-            "syllables_th.txt",
-            "thailand_provinces_th.txt",
-            "tnc_freq.txt",
-            "ttc_freq.txt",
-            "wordnet_th.db",
-            "words_th.txt",
-            "words_th_thai2fit_201810.txt",
+        "pythainlp": [
+            "corpus/*",
         ],
     },
+    include_package_data=True,
     install_requires=requirements,
     extras_require=extras,
     license="Apache Software License 2.0",