Small change in API