增加controller的支持