Fixed bug in another place