small modification
bug fixed