Cannot resolve symbol 'LargeTest'

I am converting all of my tests over to the Testing Support Library. However, when I try to import the LargeTest annotation like this

import android.support.test.filters.LargeTest;

I get Cannot resolve symbol 'LargeTest'. What dependency do I need to add to my gradle file to resolve this error?