Update: correcting a bug