Update stubs with new API