fixed little clean up bug