| @@ -129,14 +129,13 @@ | ||
| 129 | 129 | * |
| 130 | 130 | * @return string The new html code. |
| 131 | 131 | */ |
| 132 | 132 | public function addProUpsell( $old, $feature = null ) { |
| 133 | - $biz_features = array( 'schedule-chart', 'chart-permissions', 'db-query' ); | |
| 134 | 133 | $return = ''; |
| 135 | 134 | $feature = strval( $feature ); |
| 136 | - if ( empty( $feature ) || ( in_array( $feature, $biz_features ) && ! apply_filters( 'visualizer_is_business', false ) ) ) { | |
| 135 | + if ( empty( $feature ) || ( $feature == 'schedule-chart' && ! apply_filters( 'visualizer_is_business', false ) ) ) { | |
| 137 | 136 | $plan = 'PRO'; |
| 138 | - if ( in_array( $feature, $biz_features ) ) { | |
| 137 | + if ( $feature === 'schedule-chart' ) { | |
| 139 | 138 | $plan = 'BUSINESS'; |
| 140 | 139 | } |
| 141 | 140 | $return = '<div class="only-pro-content">'; |
| 142 | 141 | $return .= ' <div class="only-pro-container">'; |
