Restore missing functions