Fix context not removed when node is rerun

[lukechu10]

Closes #774

Copilot reviewed 2 out of 2 changed files in this pull request and generated no comments.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 71.41%. Comparing base (969ac04) to head (0810a55).
Report is 1 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #775      +/-   ##
==========================================
+ Coverage   71.36%   71.41%   +0.05%     
==========================================
  Files          45       45              
  Lines        6646     6658      +12     
==========================================
+ Hits         4743     4755      +12     
  Misses       1903     1903              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.