small cleanup for better looks