Fixing for build to pass