Fix unit tests for new pockets.