Fixed TODO so it's picked up by IDEs