Added assertian to test