Add type for tests