Re-enabling unit test