Minor refactoring around KryoPool