fixed a bug :)