Re-added a missing line