a small modification