fix initial bug