Get unittest working