Fixed warning issue