Restore functionality removed earlier in error