Small bugfix for recent change for #88