Change test layer to UnitTestLayer