Fixed hidden regression from previous commit