slight code clean