You could just `return s.Contains(unit)`