Fixing code style issue