#2 Fix a little bug