Warning message fix