Fix main goroutine