adding missing call to make it work properly