Fix for EE 2.8+