Fixes Edit Display bug