Fixed a bug. works now