Forgot one place for type fix