Fixed Minor Code Issue