fix edge bugs