fix funny bug :D