problem with EDIT fixed