Cannot make atom typesafe it shows as any

[khanakia]

I have this code and when i mouse hover to immerSetup.screen in vscode it shows any

How do I make it typesafe ?

type Setup = {
  screen: string;
};

const defaultSetup = {
  screen: "step1",
};

// Original atom.
const setupImmerAtom = atomWithImmer<Setup>(defaultSetup);

const setScreen = (screen: Screen) => {
  const [setupAtom, setAtomSetup] = useAtom(setupImmerAtom);
  setAtomSetup((immerSetup) => {
    immerSetup.screen = screen;
    return immerSetup;
  });
};

[rothsandro]

Works fine for me https://tsplay.dev/wRby1w

Can you create a reproduction?

[khanakia]

Now it started working fine after reloading, which is strange. I am using remix with vs code.