modify class tests