Assert against correct variable in test