correcting little bug