Warning message fixes
Warning message fixes