Fix use section

Sorry, missed use files. related to #553 and #554