fixed problem in tests. all need updating