Fix tests persisting

Close #47