Just a small modification