Minor clean up, thanks @slandelle