tiny code cleaning up