Fixing a custom columns issue