Add expect for tests