Made a little change to remove a few warnings.