Fixes edit mode styling on inputs