diff --git a/example/main.go b/example/main.go
index 572d604..cfc4362 100644
--- a/example/main.go
+++ b/example/main.go
@@ -11,6 +11,9 @@ import (
 
 func main() {
 	dir, err := os.Getwd()
+	if err != nil {
+		log.Fatal(err)
+	}
 
-	log.Fatal(http.ListenAndServe(":12345", enhancedfileserver.FileServer(http.Dir(dir))))
+	log.Fatal(http.ListenAndServe(":12345", enhancedfileserver.FileServer(enhancedfileserver.Dir(dir))))
 }