Removed expect() from unit tests because it's a hassle