diff --git a/qiskit_machine_learning/neural_networks/estimator_qnn.py b/qiskit_machine_learning/neural_networks/estimator_qnn.py
index 96696a07c..89ac0fbd1 100644
--- a/qiskit_machine_learning/neural_networks/estimator_qnn.py
+++ b/qiskit_machine_learning/neural_networks/estimator_qnn.py
@@ -183,7 +183,10 @@ def __init__(
                 self.num_virtual_qubits = circuit.num_qubits
             else:
                 circuit = pass_manager.run(circuit)
-                self.num_virtual_qubits = circuit.layout._input_qubit_count
+                if hasattr(circuit.layout, "_input_qubit_count"):
+                    self.num_virtual_qubits = circuit.layout._input_qubit_count
+                else:
+                    self.num_virtual_qubits = circuit.num_qubits
 
         self._org_circuit = circuit
 
diff --git a/qiskit_machine_learning/neural_networks/sampler_qnn.py b/qiskit_machine_learning/neural_networks/sampler_qnn.py
index b6a1eb911..ad2a17870 100644
--- a/qiskit_machine_learning/neural_networks/sampler_qnn.py
+++ b/qiskit_machine_learning/neural_networks/sampler_qnn.py
@@ -209,7 +209,10 @@ def __init__(
                 self.num_virtual_qubits = circuit.num_qubits
             else:
                 circuit = pass_manager.run(circuit)
-                self.num_virtual_qubits = circuit.layout._input_qubit_count
+                if hasattr(circuit.layout, "_input_qubit_count"):
+                    self.num_virtual_qubits = circuit.layout._input_qubit_count
+                else:
+                    self.num_virtual_qubits = circuit.num_qubits
 
         self._org_circuit = circuit