Need to type hint