Clean code
minor fix