Fixed typing error

Didn't tested the new code, probably works now.