fix check.go
small modifications