Small renaming fix :)