fixed a problem with previous commit