fix some missing stuff