correct bug in last commit