fix more edge bugs