Fixing column retrieval bug in personalized mode