Buy Now is not displayed properly

#587

Author: contactashish13

classes/Visualizer/Module/Sources.php

@@ -130,10 +130,14 @@ public function filterChartData( $data, $chart_id ) {
 	 * @return string The new html code.
 	 */
 	public function addProUpsell( $old, $feature = null ) {
+		$pro_features   = Visualizer_Module::get_features_for_license( 1 );
 		$biz_features   = Visualizer_Module::get_features_for_license( 2 );
 		$return  = '';
 		$feature = strval( $feature );
-		if ( empty( $feature ) || ( in_array( $feature, $biz_features, true ) && ! apply_filters( 'visualizer_is_business', false ) ) ) {
+		if ( empty( $feature ) ||
+			( in_array( $feature, $biz_features, true ) && ! apply_filters( 'visualizer_is_business', false ) ) ||
+			( in_array( $feature, $pro_features, true ) && ! Visualizer_Module::is_pro() )
+		) {
 			$msg = sprintf( __( 'Upgrade to %s to activate this feature!', 'visualizer' ), 'PRO' );
 			if ( Visualizer_Module::is_pro() && in_array( $feature, $biz_features, true ) ) {
 				$msg = sprintf( __( 'Upgrade your license to at least the %s version to activate this feature!', 'visualizer' ), 'DEVELOPER' );