Fixing more tests
Fixing test that uses wrongly named variable