Trying to fix these units tests.. not idea why they aren't working.