fix initial code