Docker build fix