(Minor) Fix to partial updating