get tests passing again