minor change

change to according to data structure