Fixing test code
Was trying to brute-force grab the context.