fix type assert in test