fixed a bug :D