Fixed small bugje