Fix uclearn with setup.py

uclearn was broken when used with setup.py