Fixing a warning message