fixing count input missing