Fixed unit test after Ash's fix