Fixed small error in previous commit