Fix deleted code