Improve code style (small)