Update allocate tests for full code coverage