getting tests working again