minor tweak to the code to make it look cleaner