Add missing @VisibleForTesting annotation