Fixing error made in last commit