Fixed tiny bug