super minor code refactor