assign class tests passing again