fixed a todo, yay :D