Small bit of code cleaning