Fixed TypeError from previous commit