Fix parameter passing to unit