Properly fixed minor bug