Fixes sample after merge broke API