fix end bugs