diff --git a/src/filters.php b/src/filters.php
index d8c7795..e65567d 100644
--- a/src/filters.php
+++ b/src/filters.php
@@ -49,5 +49,5 @@ add_filter('template_include', function ($main) {
     if (!is_string($main) || !(string) $main) {
         return $main;
     }
-    return template_wrap(new Wrapper(basename($main)));
+    return template_wrap(new Wrapper($main));
 }, 109);
diff --git a/src/lib/Sage/Template/Wrapper.php b/src/lib/Sage/Template/Wrapper.php
index 4c291cc..9440816 100644
--- a/src/lib/Sage/Template/Wrapper.php
+++ b/src/lib/Sage/Template/Wrapper.php
@@ -60,6 +60,6 @@ class Wrapper implements WrapperInterface
     public function getTemplate()
     {
         $template = apply_filters('sage/unwrap_' . $this->slug, $this->template) ?: $this->template;
-        return locate_template($template);
+        return locate_template($template) ?: $template;
     }
 }