Fix missing change from prev commit