Made edits in working to make the test work.