Fix build error(test)