Restore real time updates functionality