Small TODO change