fixes so unitttest works