Fix go.net/context import