Fix a few edit mode bugs (css)