Fixing warning message