fixed bug for restart game