fixed up the TODO item