Fix up logging

Logging returns unit, instead of wrapping an expression