diff --git a/faker_test.go b/faker_test.go
index 254fe5d..60018b8 100644
--- a/faker_test.go
+++ b/faker_test.go
@@ -3,9 +3,9 @@ package faker
 import (
 	"fmt"
 	"math/rand"
+	"reflect"
 	"strings"
 	"testing"
-	"reflect"
 )
 
 func Expect(t *testing.T, expected, got interface{}, values ...interface{}) {
@@ -191,4 +191,4 @@ func TestBoolWithChance(t *testing.T) {
 	Expect(t, false, f.BoolWithChance(0))
 	Expect(t, true, f.BoolWithChance(101))
 	Expect(t, false, f.BoolWithChance(-1))
-}
\ No newline at end of file
+}
diff --git a/file_test.go b/file_test.go
index b7cd4fe..43daf44 100644
--- a/file_test.go
+++ b/file_test.go
@@ -7,7 +7,7 @@ import (
 
 func TestExtension(t *testing.T) {
 	p := New().File()
-	Expect(t, true, len(p.Extension()) > 2)
+	Expect(t, true, p.Extension() != "")
 }
 
 func TestFileWithExtension(t *testing.T) {