bug fix on edit module