Fixed a little but crucial bug