Fixed wrong include and comment