Fixing little stupid bug