Fixed small code issue.