Fix test stub