Fixes regression caused by #124