diff --git a/dcc/config.py b/dcc/config.py
index 7fba22b..e586752 100644
--- a/dcc/config.py
+++ b/dcc/config.py
@@ -2,7 +2,7 @@ import io
 import pathlib
 import os
 import re
-import tomlkit
+import tomlkit.toml_file
 
 from cliff.command import Command