diff --git a/extensions/ql-vscode/src/view/method-modeling/MethodModelingView.tsx b/extensions/ql-vscode/src/view/method-modeling/MethodModelingView.tsx index ea9c9a9b2..95cc357c5 100644 --- a/extensions/ql-vscode/src/view/method-modeling/MethodModelingView.tsx +++ b/extensions/ql-vscode/src/view/method-modeling/MethodModelingView.tsx @@ -10,6 +10,7 @@ import { vscode } from "../vscode-api"; import { NotInModelingMode } from "./NotInModelingMode"; import { NoMethodSelected } from "./NoMethodSelected"; import { MethodModelingPanelViewState } from "../../model-editor/shared/view-state"; +import { convertFromLegacyModeledMethod } from "../../model-editor/shared/modeled-methods-legacy"; type Props = { initialViewState?: MethodModelingPanelViewState; @@ -31,7 +32,10 @@ export function MethodModelingView({ initialViewState }: Props): JSX.Element { const modelingStatus = useMemo( () => - getModelingStatus(modeledMethod ? [modeledMethod] : [], isMethodModified), + getModelingStatus( + convertFromLegacyModeledMethod(modeledMethod), + isMethodModified, + ), [modeledMethod, isMethodModified], ); @@ -95,7 +99,7 @@ export function MethodModelingView({ initialViewState }: Props): JSX.Element { diff --git a/extensions/ql-vscode/src/view/method-modeling/ModeledMethodsPanel.tsx b/extensions/ql-vscode/src/view/method-modeling/ModeledMethodsPanel.tsx index 824a10efd..a47ec9cd7 100644 --- a/extensions/ql-vscode/src/view/method-modeling/ModeledMethodsPanel.tsx +++ b/extensions/ql-vscode/src/view/method-modeling/ModeledMethodsPanel.tsx @@ -4,6 +4,7 @@ import { MethodModelingInputs } from "./MethodModelingInputs"; import { Method } from "../../model-editor/method"; import { styled } from "styled-components"; import { MultipleModeledMethodsPanel } from "./MultipleModeledMethodsPanel"; +import { convertToLegacyModeledMethod } from "../../model-editor/shared/modeled-methods-legacy"; export type ModeledMethodsPanelProps = { method: Method; @@ -26,9 +27,7 @@ export const ModeledMethodsPanel = ({ return ( 0 ? modeledMethods[0] : undefined - } + modeledMethod={convertToLegacyModeledMethod(modeledMethods)} onChange={onChange} /> );