Update unit tests to classy beta