fix failing unit test

Fixed test failing due to incomplete data structure and wrong variable name