Fixed one minor bug