small change - tidy up