fix the hard coding number bug