diff --git a/plugins/wpe-headless/includes/replacement/callbacks.php b/plugins/wpe-headless/includes/replacement/callbacks.php
index a40e3fa0e..41448949c 100644
--- a/plugins/wpe-headless/includes/replacement/callbacks.php
+++ b/plugins/wpe-headless/includes/replacement/callbacks.php
@@ -130,7 +130,13 @@ function wpe_headless_post_preview_link( $link, $post ) {
 		/**
 		 * Add preview and preview ID back to path to support Next.js preview mode
 		 */
-		$path = trailingslashit( $path ) . 'preview/' . $preview_id;
+		if ($post->post_status !== 'publish') {
+			$path = 'draft/';
+		} else {
+			$path = $path ? trailingslashit( $path ) : '';
+		}
+
+		$path .= 'preview/' . $preview_id;
 
 		$link = add_query_arg(
 			array(