fix type, thank god for unit tests