Fixed invalid variable in unitest