Fix error cuz of active display

Fix for Andy's error