diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
index 9d8217f9cd8f6..116cafcc81797 100644
--- a/tests/trainer/test_trainer.py
+++ b/tests/trainer/test_trainer.py
@@ -30,7 +30,7 @@
 def test_model_pickle(tmpdir):
     import pickle
 
-    model = TestModelBase()
+    model = TestModelBase(tutils.get_default_hparams())
     pickle.dumps(model)