Fixed issue causing build to fail