Fix minor bug: