diff --git a/pg_tde--1.0.sql b/pg_tde--1.0.sql
index 8a5cd5d6..e0646dc5 100644
--- a/pg_tde--1.0.sql
+++ b/pg_tde--1.0.sql
@@ -165,7 +165,8 @@ SELECT EXISTS (
     SELECT 1
     FROM   pg_catalog.pg_class
     WHERE  relname = table_name
-    AND    relam = (SELECT oid FROM pg_catalog.pg_am WHERE amname = 'tde_heap_basic')
+    AND    (relam = (SELECT oid FROM pg_catalog.pg_am WHERE amname = 'tde_heap_basic')
+        OR relam = (SELECT oid FROM pg_catalog.pg_am WHERE amname = 'tde_heap'))
     )$$
 LANGUAGE SQL;