Added types for test