Update module to handle OL6