fix minor isssue