Add missing line back in