diff --git a/sandbox/test_leave.py b/sandbox/test_leave.py
index 52715ec..2094bfd 100644
--- a/sandbox/test_leave.py
+++ b/sandbox/test_leave.py
@@ -25,7 +25,7 @@
 from webnotes.model.doc import Document
 from webnotes.model.code import get_obj
 from webnotes.utils import cstr, flt
-from webnotes.model.doclist import getlist
+from webnotes.model.wrapper import getlist
 sql = webnotes.conn.sql
 
 from sandbox.testdata import leaves