Small code styel fix