Bug fix

Forgot type