Small refactor code style