Fix code to call the method instead pass the function