Fix container build