Fix error causing feature to fail