Set unit not working, wip