Fixed minor error in last commit