Change test to integration as it sometimes fails