Clean up display flex detection