fixed todo bug