Update test to use proper assert for tests