Fixed imports

In main.go