Minor todo change