(Minor) Style improvement