fixed bug in last commit