Small todo change