Small cleanup :3