Type error fix

Hopefully, it works locally.