diff --git a/capture/af_packet_linux.go b/capture/af_packet_linux.go
index 9fc8a9b..53debc8 100644
--- a/capture/af_packet_linux.go
+++ b/capture/af_packet_linux.go
@@ -1,13 +1,10 @@
 // +build linux
 
-package main
+package capture
 
 import (
 	"flag"
 	"fmt"
-	"log"
-	"os"
-	"runtime/pprof"
 	"time"
 
 	"github.com/google/gopacket"