small cleanup

for more consistency in the code