Fixed the test task to also retore build assets