Fix canceling unit test