Fix for go cover (again)