Fix Square Edit fail bug