Fixed bug in akit