Fix robolectric tests