Made the code look better, no functional change