fixing unit test (will need updating once Chris looks again at the pruning