diff --git a/ssd1306/example_test.go b/ssd1306/example_test.go
index ee000c4..4f6feda 100644
--- a/ssd1306/example_test.go
+++ b/ssd1306/example_test.go
@@ -5,6 +5,7 @@
 package ssd1306_test
 
 import (
+    "flag"
 	"image"
 	"log"