[ToDo]Tweaks

silly mistake from previous commit