small update & fix