Fix TypeError

Signed-off-by: MiLk <hello@emilienkenler.com>