diff --git a/x/staking/keeper/keeper_test.go b/x/staking/keeper/keeper_test.go
index dbde12e7b..8b59555e6 100644
--- a/x/staking/keeper/keeper_test.go
+++ b/x/staking/keeper/keeper_test.go
@@ -1,24 +1,28 @@
-package keeper
+package keeper_test
 
 import (
 	"testing"
 
 	"github.com/stretchr/testify/require"
 
+	"github.com/cosmos/cosmos-sdk/simapp"
 	"github.com/cosmos/cosmos-sdk/x/staking/types"
+	abci "github.com/tendermint/tendermint/abci/types"
 )
 
 func TestParams(t *testing.T) {
-	ctx, _, _, keeper, _ := CreateTestInput(t, false, 0)
+	app := simapp.Setup(false)
+	ctx := app.BaseApp.NewContext(false, abci.Header{})
+
 	expParams := types.DefaultParams()
 
 	//check that the empty keeper loads the default
-	resParams := keeper.GetParams(ctx)
+	resParams := app.StakingKeeper.GetParams(ctx)
 	require.True(t, expParams.Equal(resParams))
 
 	//modify a params, save, and retrieve
 	expParams.MaxValidators = 777
-	keeper.SetParams(ctx, expParams)
-	resParams = keeper.GetParams(ctx)
+	app.StakingKeeper.SetParams(ctx, expParams)
+	resParams = app.StakingKeeper.GetParams(ctx)
 	require.True(t, expParams.Equal(resParams))
 }