Refactor: change set className to jsonKey