Minor changes to register pool implementation