fixed some missing stuff