add missing line in code