Fix more todos