diff --git a/app/controller/class-ms-controller-plugin.php b/app/controller/class-ms-controller-plugin.php
index 60ad9daa..0a69a5ee 100644
--- a/app/controller/class-ms-controller-plugin.php
+++ b/app/controller/class-ms-controller-plugin.php
@@ -961,9 +961,10 @@ public function custom_single_template( $default_template ) {
 	 *
 	 * @return string The custom template path.
 	 */
+	
 	public function custom_page_template( $default_template ) {
 		$template = '';
-
+	
 		// Checks for invoice single template.
 		if ( $type = MS_Model_Pages::is_membership_page() ) {
 			$membership_id = apply_filters(
@@ -972,26 +973,25 @@ public function custom_page_template( $default_template ) {
 				true,
 				true
 			);
-
+	
 			if ( $membership_id ) {
 				$template = get_query_template(
 					'm2',
-					'm2-' . $type . '-' . $membership_id . '.php'
+					array('m2-' . $type . '-' . $membership_id . '.php')
 				);
 			}
-
+	
 			if ( ! $template ) {
 				$template = get_query_template(
 					'm2',
-					'm2-' . $type . '.php'
+					array('m2-' . $type . '.php')
 				);
 			}
 		}
-
+	
 		if ( ! $template ) {
 			$template = $default_template;
 		}
-
 		return $template;
 	}