Fix for update