Fix test.py

Fixes usage of non-existing variable

Change-Id: Ie34a8606c209c77f5f8299d3e25ba594c772520d