Fix a bug

Forgot about that, oops :p