Fix missing line in last commit