fixing integration Test