fixed a bug
small addition