Corrected minor bug