Made a little change