fix some forgotten stuff