Android Espresso testing 'Cannot resolve symbol 'InstrumentationRegistry''

I'm trying to import

 import android.support.test.InstrumentationRegistry;

my build.gradle file

androidTestCompile 'com.android.support.test:testing-support-lib:0.1'
androidTestCompile 'com.android.support.test:runner:0.2'
androidTestCompile 'com.android.support.test:rules:0.2'
androidTestCompile 'com.android.support.test.espresso:espresso-core:2.2.2'

in default config:

defaultConfig {
        testInstrumentationRunner "android.support.test.runner.AndroidJUnitRunner"
    }

Is there a library I'm missing here? I'm trying to import InstrumentationRegistry but it doesn't recognise it!