Fix issue where tests where test failed when no context was set