small change to set the calling of the function correctly