diff --git a/demo.m b/demo.m
index 6f3ddaf..d390108 100755
--- a/demo.m
+++ b/demo.m
@@ -53,11 +53,11 @@
     
     
     %% perform fast hybrid precoding algorithm
-    [FRFM, FBBM] = fast_hybrid_precoding_wrapper_OFDM( Fopt, NRF, 0 );
+    [FRFM, FBBM] = fast_hybrid_precoding_OFDM_wrapper( Fopt, NRF, 0 );
     for l = 1:L
         FBBM(:,:,l) = sqrt(Ns) * FBBM(:,:,l) / norm(FRFM * FBBM(:,:,l),'fro');
     end
-    [WRFM, WBBM] = fast_hybrid_precoding_wrapper_OFDM( Wopt, NRF, 0 );
+    [WRFM, WBBM] = fast_hybrid_precoding_OFDM_wrapper( Wopt, NRF, 0 );
 
     
     %% Calculate the spectral efficiency
@@ -85,4 +85,4 @@
 set(ax1,'FontSize',fs);
 xlabel('SNR [dB]')
 ylabel('Spectral efficiency (bits/s/Hz)')
-legend('Fast hybrid precoding algorithm', 'Location', 'northwest')
\ No newline at end of file
+legend('Fast hybrid precoding algorithm', 'Location', 'northwest')