Fix comments for #204