fix end bug