Minor code clean