Fix for if unit function