Small alteration

Tidy