Fix some unit tests to properly test the code

Added a .toBe() statement to all the expect() calls that were missing them. This will ensure that the code is properly tested.