Fixed som minor bug