Fixed small error with last commit