Minor layout correction