correcting minor bug