Fix simple bug