Fixed a TODO for test

Don't make such a mistake again.