Restore functionality