Fix variable name
Update TODO