Minor UI fix on the builder