Added missing include and fixed syntax error