Fixed call to wrong function. Related to #4.