Fix Mismatch in neuron count