Small update. 

fixed: wrong function called.