slight update of end to make fit