Added temporary fix for edit image