small modification, bug in file