Fixed small mistake