FIx regression from previous commit