Fixed a mini bug