Update extract_test.go

Cleaned up import juggling.