diff --git a/mode_test.go b/mode_test.go
index bf6a5a1..0a9cbc3 100644
--- a/mode_test.go
+++ b/mode_test.go
@@ -12,10 +12,14 @@ import (
 )
 
 func init() {
-	SetMode(TestMode)
+	os.Setenv(ENV_GIN_MODE, TestMode)
 }
 
 func TestSetMode(t *testing.T) {
+	assert.Equal(t, ginMode, testCode)
+	assert.Equal(t, Mode(), TestMode)
+	os.Unsetenv(ENV_GIN_MODE)
+
 	SetMode(DebugMode)
 	assert.Equal(t, ginMode, debugCode)
 	assert.Equal(t, Mode(), DebugMode)
@@ -29,9 +33,4 @@ func TestSetMode(t *testing.T) {
 	assert.Equal(t, Mode(), TestMode)
 
 	assert.Panics(t, func() { SetMode("unknown") })
-
-	os.Setenv(ENV_GIN_MODE, DebugMode)
-	assert.Equal(t, ginMode, debugCode)
-	assert.Equal(t, Mode(), DebugMode)
-	os.Unsetenv(ENV_GIN_MODE)
 }