Fix wrong assert in test