#41 Fix test regression