diff --git a/src/Extension/FluentCMSMainExtension.php b/src/Extension/FluentCMSMainExtension.php
index 318f13c2..71832e7a 100644
--- a/src/Extension/FluentCMSMainExtension.php
+++ b/src/Extension/FluentCMSMainExtension.php
@@ -106,7 +106,7 @@ public function publish_localised_copy($data, $form)
     protected function getRecordForLocalisedAction($data, $form): ?DataObject
     {
         $id = (int) $data['ID'];
-        $className = $this->owner->config()->get('model_class');
+        $className = $this->owner->getModelClass();
 
         if (!$id || !$className) {
             // Invalid inputs
diff --git a/src/Middleware/LocaleSwitchRedirector.php b/src/Middleware/LocaleSwitchRedirector.php
index dc3399b1..c18c208a 100644
--- a/src/Middleware/LocaleSwitchRedirector.php
+++ b/src/Middleware/LocaleSwitchRedirector.php
@@ -32,6 +32,7 @@ public function process(HTTPRequest $request, callable $delegate)
         if ($response->getStatusCode() === 404
             && !$state->getIsFrontend()
             && $this->getParamLocale($request)
+            && !$request->getVar('CMSPreview')
         ) {
             // Redirect to the CMS home page if the requested page doesn't exist
             $response = new HTTPResponse();