fixed tiny todo

(also, fixed some types --- thanks NikosG)