Fixed action tests to handle context passing.