Unit test restored