Script fix
Fixed unit test