Removed TODO from sample / test file.