minor bug fixed
ready for production