diff --git a/tests/lean/uni_bug1.lean b/tests/lean/uni_bug1.lean
index 643e370b7..098c846cd 100644
--- a/tests/lean/uni_bug1.lean
+++ b/tests/lean/uni_bug1.lean
@@ -1,4 +1,4 @@
-import data.nat data.prod
+import data.nat.basic data.prod
 open nat prod
 
 variable R : nat → nat → Prop
diff --git a/tests/lean/uni_bug1.lean.expected.out b/tests/lean/uni_bug1.lean.expected.out
index dcd7edee7..9cadb966b 100644
--- a/tests/lean/uni_bug1.lean.expected.out
+++ b/tests/lean/uni_bug1.lean.expected.out
@@ -1,2 +1 @@
-uni_bug1.lean:2:0: warning: imported file uses 'sorry'
 f 1 0 (Rtrue (pr1 (pair 1 0)) 0) : nat