small update and fix