fixed minor bug introduced in last commit