Fix PIE Control logic?