Fix to my prev commit