Small api change