Import the log function from ufl as well.