Removed unnecessary include and fixed silly mistake.