Fix deployment unit tests