diff --git a/pandas/core/frame.py b/pandas/core/frame.py
index a420266561c5a..24e344fa9870c 100644
--- a/pandas/core/frame.py
+++ b/pandas/core/frame.py
@@ -2839,9 +2839,9 @@ def query(self, expr, inplace=False, **kwargs):
 
         Examples
         --------
-        >>> from numpy.random import randn
-        >>> from pandas import DataFrame
-        >>> df = pd.DataFrame(randn(10, 2), columns=list('ab'))
+        >>> import numpy as np
+        >>> import pandas as pd
+        >>> df = pd.DataFrame(np.random.randn(10, 2), columns=list('ab'))
         >>> df.query('a > b')
         >>> df[df.a > df.b]  # same result as the previous expression
         """