diff --git a/dkim/tests/test_dkim_generate.py b/dkim/tests/test_dkim_generate.py
index 1649ab5..b046436 100644
--- a/dkim/tests/test_dkim_generate.py
+++ b/dkim/tests/test_dkim_generate.py
@@ -22,7 +22,7 @@ import tempfile
 import unittest
 
 import dkim
-import dknewkey
+import dkim.dknewkey as dknewkey
 
 def read_data(path):
     """Get the content of the given test data file."""