Fix to warning message