fixed funny bug :)