Fix Docker support