Adds back missing Group import