Change for bug

Just some modification