CSS flex correction