Add TODO to shouldLog