Tweak repl context test

Tests are failing on CI. Now using a different approach to test the same outcome