Fixed: Use JSON, not AST