HOT FIX for minor bug

Hi, it is a quick fix for minor miostake. So I will do a self merge to save time.
Tack :)