fixed di bug