Fixed a mini-bug