remove the temp class for test