remove function for tests is now working