Then just do a final selem[selem > 0] = 1