Fixed bug stopping layouts getting set