Restore accidentally deleted code