minor bug fix ) missed