Got sample test working again