Corrected minor bug introduced by last commit