Fixing a warning issue