fixed a bug with teh previous commit