minor bug corrected - diff