Fixed a todo