small bug fix - thanks Lorenzo!