Fixed bug (minor)