Restore missing code