Add tests for charge group restriction