Update tests to match style classes