fix broken unit tests. all working now.