Fixed warning and added dep