Fix one more type assumption error