fix inintial code