Remove unused import in gala-train