diff --git a/examples/regression.py b/examples/regression.py
index 50ed0dc..735d47e 100644
--- a/examples/regression.py
+++ b/examples/regression.py
@@ -1,6 +1,9 @@
 from sklearn.metrics import mean_squared_error
 from sklearn.model_selection import train_test_split
 
+import numpy as np
+import pandas as pd
+
 from ngboost import NGBRegressor
 from ngboost.distns import Normal