fixed a bug ... maybe