Missed one small change