Fix todo