Fix small adjustment