get tests working again