add TestOnly keyword