Fix tiny bug (previous commit)