Fix for updating