Adding missing call to a function