Fix/ typing error