diff --git a/src/index.ts b/src/index.ts
index 60761b0..db67166 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -118,7 +118,7 @@ export default function PluginVue(userOptions: Partial<Options> = {}): Plugin {
               ? descriptor.script!
               : query.type === 'style'
               ? descriptor.styles[query.index]
-              : query.type === 'custom'
+              : filterCustomBlock(query.type)
               ? descriptor.customBlocks[query.index]
               : null