Fix incomplete change which was breaking the build