Python format on save broken

Currently it seems that the editor is not updating after formatOnSave: true for python files when using Black. This works fine in VS code.

The file is saved, but the editor is not updated with the formatted file. So next time you try to save it says the files are out of sync and doesn’t save.

Note that isort does seem to work here and update the editor, but not formatting with Black.

Here’s my config:

  "[python]": {
    "editor.defaultFormatter": "mikoz.black-py",
    "editor.tabSize": 4,
    "editor.codeActionsOnSave": {
      "source.organizeImports": true
    },
    "editor.formatOnSave": true
  },
  "isort.args": ["--profile", "black"]

Anyone have any luck with this?