Fix missing line (oops)