FIXed it quick

comments #1 is to make the fix better than this.