fixup! Fix code to call the method instead pass the function