Fixed small errors in last commit