correct a little bug