Fixed api for using yaml driver