Fix error causing build to fail