fixed edit bug ...?