| @@ -129,15 +129,15 @@ | ||
| 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' ); | |
| 133 | + $biz_features = array( 'schedule-chart', 'chart-permissions', 'db-query' ); | |
| 134 | 134 | $return = ''; |
| 135 | 135 | $feature = strval( $feature ); |
| 136 | - if ( empty( $feature ) || ( in_array( $feature, $biz_features ) && ! apply_filters( 'visualizer_is_business', false ) ) ) { | |
| 136 | + if ( empty( $feature ) || ( in_array( $feature, $biz_features, true ) && ! apply_filters( 'visualizer_is_business', false ) ) ) { | |
| 137 | 137 | $plan = 'PRO'; |
| 138 | - if ( in_array( $feature, $biz_features ) ) { | |
| 139 | - $plan = 'BUSINESS'; | |
| 138 | + if ( in_array( $feature, $biz_features, true ) ) { | |
| 139 | + $plan = 'DEVELOPER'; | |
| 140 | 140 | } |
| 141 | 141 | $return = '<div class="only-pro-content">'; |
| 142 | 142 | $return .= ' <div class="only-pro-container">'; |
| 143 | 143 | $return .= ' <div class="only-pro-inner">'; |
| @@ -146,9 +146,9 @@ | ||
| 146 | 146 | $return .= ' </div>'; |
| 147 | 147 | $return .= ' </div>'; |
| 148 | 148 | $return .= '</div>'; |
| 149 | 149 | } |
| 150 | - if ( empty( $feature ) && defined( 'VISUALIZER_PRO_VERSION' ) ) { | |
| 150 | + if ( empty( $feature ) && VISUALIZER_PRO ) { | |
| 151 | 151 | remove_filter( 'visualizer_pro_upsell', 'addProUpsell', 10, 1 ); |
| 152 | 152 | $return = ''; |
| 153 | 153 | } |
| 154 | 154 | |
