Fixed bug with edit mode displaying in wrong column