got tests working again