Fix bug di