FIX expit bug with out != None