Fix as per @samjsharpe's comment