fixed smaller bug