Fix master

Fixes the regression caused by #12