Fixed the warning