Minor issue fix