Fixed up test functionality