Fixing error in last commit