Fixing test to use shade variable which is cleaned up after the test