Fix todo test