Update tests to work with go version