layout fix
little change agin