Add assert to test