Small fix of a bug i caused.