fix type in test :(