Fix for transit feature in plot