Correcting a little bug