Fixed a bug. Ma bad