fixed a bug. yeah