Fix error as pointed out by @antgonza