Fixed code found by failing test due to change in builder.