Fix Unit deployment