Fix after renaming variable