Fix docummentation comment (delete TODO)