Add layout fix