Fix context of remove call on rules