Small cleanup fixe for error