update: Fix critical bug