cleaner fix

made a cleaner fix