Fixed a bug

Thanks!