fixed the TODO