Minor Bug Correction